1.绘制简单的图形
import numpy as np
import matplotlib.pyplot as plt
#生成数据
x = np.arange(0, 6, 0.1)
y = np.sin(x)
#绘制图形
plt.plot(x, y)
plt.show()
首先通过NumPy中的arange方法生成了从0到6,间隔为0.1的数据,设置成了x。并且对其进行sin计算,生成了相同数量的y,最终将x,y的数据传给了plt.plot绘制出图形,然后通过plt.show显示图形。如图所示。
2.pyplot的功能
pyplot当然也可以同时绘制多个图形,并且可以在图上添加注释。
import numpy as np
import matplotlib.pyplot as plt
x = np.arange(0,6,0.1)
y1 = np.sin(x)
y2 = np.cos(x)
plt.plot(x, y1, label="sin")
plt.plot(x, y2, linestyle = "--", label="cos")
plt.xlabel("x")
plt.ylabel("y")
plt.title('sin & cos')
plt.legend()
plt.show()
效果如下图所示
在这段代码中,label代表的是线的标签,linestyle代表线的类型。可以见下表:
类型 | 简写 | 说明 |
---|---|---|
solid(默认) | “-” | 实线 |
dotted | “.” | 点虚线 |
dashed | “–” | 破折线 |
dashdot | “-.” | 点划线 |
none | “”或者“ ” | 不划线 |
xlabel和ylabel分别代表x轴标签和y轴标签,title代表标题。
更多详情参考:Matplotlib 教程
3.显示图片
pyplot还可以显示图像,通过imread函数。