一、安装拓展包
pip install matplotlib
二 、折线图
import numpy as np
import matplotlib.pyplot as plt
x = np.linspace(0, 2, 100)#创建等差数列 0-2之间100个
plt.plot(x, x, label="line1")#第一个参数为横坐标 第二个为纵坐标 第三个为曲线名字
plt.plot(x, x**2, label="line2")
plt.plot(x, x**3, label="line3")
plt.xlabel("x label")#x轴名字
plt.ylabel("y label")#y轴名字
plt.title("测试折线图")#图标名字
plt.legend()#显示图例
plt.show()#生成图表
三 、散点图
import numpy as np
import matplotlib.pyplot as plt
x = np.arange(0., 5., 0.2)#从0-5之间 ，产生等差数列 ，公差为0.2
plt.plot(x, x, "r--", x, x**2, "bs", x, x**3, "g^")#
plt.xlabel("x label")#x轴名字
plt.ylabel("y label")#y轴名字
plt.title("测试折线图")#图标名字
plt.legend()#显示图例
plt.show()#生成图表
四、三维散点图
import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
data = np.random.randint(0, 255, size=[40,40,40])#产生0~255的随机数40x3个
x,y,z = data[0],data[1],data[2]#x y z分别40个随机数
ax = plt.subplot(111, projection="3d")#创建三维绘图
ax.scatter(x[:10], y[:10], z[:10], c="y")#绘制散点
ax.scatter(x[10:20], y[10:20], z[10:20], c="r")
ax.scatter(x[30:40], y[30:40], z[30:40], c="g")
ax.set_xlabel("X")#x轴名字
ax.set_ylabel("Y")#y轴名字
ax.set_zlabel("Z")#z轴名字
plt.title("测试折线图")#图标名字
plt.legend()#显示图例
plt.show()#生成图表
五 、三维平面图
import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
fig = plt.figure()#创建一个新的画板
ax = Axes3D(fig)
X = np.arange(-4, 4, 0.25)#产生-4~4之间的等差数列，公差0.25
Y = np.arange(-4, 4, 0.25)
X,Y = np.meshgrid(X, Y)#画网格
R = np.sqrt(X**2 + Y**2)
Z = np.sin(R)#产生z值
ax.plot_surface(X, Y, Z, rstride=1, cstride=1, cmap="rainbow")#绘图
plt.show()#生成图表
