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=点型)
结果显示