matplotlib是一个Python的2D绘图库,它能够创建线图、柱状图、散点图等静态、动态和交互式的图像。matplotlib的设计目标是使用户能够在脚本语言中方便地制作出版质量级别的图表。
matplotlib可以生成各种硬拷贝格式(如PDF)以及交互式环境下的图形窗口。它也可以非常方便地定制每一个元素的属性,包括线型、颜色、标签等。
使用matplotlib的基本步骤如下:
1. 导入模块:首先需要导入matplotlib.pyplot模块,这是一个面向对象的接口,提供了很多绘图方法。
```python
import matplotlib.pyplot as plt
```
2. 创建图像:通过plt.figure()函数创建一个新的图像窗口。
```python
fig = plt.figure()
```
3. 创建子图:在图像窗口中添加一个或多个子图,可以通过add_subplot()函数实现。
```python
ax = fig.add_subplot(1, 1, 1)
```
4. 绘制数据:调用相应的绘图方法在子图上绘制数据,例如plot()函数用于绘制折线图。
```python
x = [1, 2, 3, 4]
y = [10, 20, 30, 40]
ax.plot(x, y)
```
5. 显示图像:调用show()函数显示图像。
```python
plt.show()
```
以上就是使用matplotlib进行基本绘图的基本步骤。除此之外,matplotlib还提供了丰富的API供用户定制图像的各种属性,例如设置标题、标签、图例、颜色、线型等。