Python matplotlib绘图

python绘图

 

使用python可以很轻易的对一些简单的图进行绘制，不仅可以对二维的图，而且可以绘制三维图，代码简单易懂。

在使用python进行绘图时，需要调用 matplotlib 和 numpy 包，在调包时使用以下语句

 

from matplotlib import pyplot as plt
import numpy as np

如果需要绘制三维图，则还需调用 mpl_toolkits.mplot3d

from mpl_toolkits.mplot3d import Axes3D

plt 和 np 是创建的对象，方便后面使用包中的属性

 

1、绘制二维图

 

绘制 sin 和 cos 函数的图像

 

 

from matplotlib import pyplot as plt
import numpy as np

a = np.linspace(0, np.pi*2, 50)  # 表示 a 在 0-2*PI 之间有 50 个取值的点

b = np.sin(a)   # sin 和 cos 都是 numpy 中的函数，使用对象 np 进行调用

c = np.cos(a)

 

plt.plot(a, b, color='red')   # plot 是 matplotlib 中的工具，使用对象 plt 进行调用，使用参数 color 可以设置图线的颜色

plt.plot(a, c, color='green')

plt.show()        # show 显示出图像

 

结果显示

 

 

2、绘制三维图

 

绘制曲面

 

from matplotlib import pyplot as plt
import numpy as np
from mpl_toolkits.mplot3d import Axes3D

fig = plt.figure()
ax = Axes3D(fig)

X = np.arange(-2, 2, 0.1)
Y = np.arange(-2, 2, 0.1)
X, Y = np.meshgrid(X, Y)
Z = X**2 + Y**2

ax.plot_surface(X, Y, Z, rstride=1, cstride=1, cmap='rainbow')   # rstrid 和 cstrid 分别表示行距和列距，cmap 表示颜色映射

plt.show()

 

三维曲面：ax.plot_surface(X, Y, Z, rstride=1, cstride=1, cmap='rainbow')

三维线框： ax.plot_wireframe(x, y, z, rstride=行距, cstrid=列距, linewidth=线宽, color=颜色)

三维散点： ax.scatter(x, y, z, s=大小, c=颜色, marker=点型)

 

结果显示