绘制正弦波
#绘制正弦波
import numpy as np
x = np.arange(0, 2 * np.pi, 0.1)
y = np.sin(x)
plt.title("正弦波")
plt.plot(x, y)
plt.show()
arange()是在(0,2π)周期内选取步长值为0.1作为正弦函数的x坐标,y的值则是由numpy中的sin函数。
绘制多个图像
在绘制正弦波的基础上,我们再绘制一条余弦波。通过subplot()函数来使两幅图像放在一张图上。
- subplot(x,y,z)中的实参作用如下
- x代表有几行
- y代表有几列
- z代表是第几幅图
- 此图可以直观了解该参数的作用
#绘制正弦波
import numpy as np
x = np.arange(0, 2 * np.pi, 0.1)
y1 = np.sin(x)
y2 = np.cos(x)
#绘制第一个图像
plt.subplot(2,1,1)
plt.title("正弦波")
plt.plot(x, y1)
#绘制第二个图像
plt.subplot(2,1,2)
plt.title("余弦波")
plt.plot(x, y2)
plt.show()
添加图例
在一幅图中有多条线段时,我们需要添加标签来分别各段图像,只需在plot()中添加参数‘label = ’,再调用legend()就可以了
- 代码如下:?
import numpy as np
x = np.arange(0, 2 * np.pi, 0.1)
y1 = np.sin(x)
y2 = np.cos(x)
plt.plot(x, y1, label = "sin")
plt.plot(x, y2, label = "cos")
plt.legend()
plt.show()
生成条形图
bar()函数用来生成条形图。bar()的使用方法与plot()类似,因此不再赘述。
代码如下:
# 绘制条形图
x = [2005, 2006, 2007, 2008, 2009]
y = [12, 16, 17, 26, 35]
plt.xlabel("年份")
plt.ylabel("人数")
plt.bar(x, y, align= 'center')
plt.title("条形图")
plt.show()