matplotlib是款比较基础的绘图工具。本文简单地介绍它基本的使用。
首先要导入包matplotlib.pyplot:
import matplotlib.pyplot as plt
import numpy as np
% matplotlib inline
颜色、标签、线型
往plot( )里传入要画处的变量外,还可以传入设置颜色,标签,线型的参数,分别是color, marker, linestyle.
# 创建有两个画布的画板
fig, axes = plt.subplots(2)
# 在第一个画布0到100的均匀分布的50个随机数,红色,圆形,虚线
axes[0].plot(np.random.randint(0, 100, 50), 'ro--')
# 上面的写法等价于:
axes[1].plot(np.random.randint(0, 100, 50), color='r', linestyle='dashed', marker='o')
颜色,标签,线型的参数还有:
刻度、标签、图例
# 创建一个画布
fig, ax = plt.subplots(1)
ax.plot(np.random.randn(1000).cumsum())
# 设置刻度
#plt.xlim([0,500])
ax.set_xlim([0, 800])
# 设置显示的刻度
#plt.xticks([0,500])
ax.set_xticks(range(0,500,100))
# 设置刻度标签
ax.set_yticklabels(['Jan', 'Feb', 'Mar'])
# 设置坐标轴标签
ax.set_xlabel('Number')
ax.set_ylabel('Month')
# 设置标题
ax.set_title('Example')
# 图例
ax.plot(np.random.randn(1000).cumsum(), label='line1')
ax.plot(np.random.randn(1000).cumsum(), label='line2')
ax.legend(loc='best')
plt.legend()