04_1Python绘图_matplotlib

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()