【Python】图像绘制

【Python】图像绘制

• 二维图像
• 三维图像
• 散点图绘制

二维图像

三维图像

图像方程：z = xx+yy

绘制三维图像我们需要用到Axes3D库，调用ax.plot_surface进行绘制，其中plot_surface的参数如下图所示

参数	解释
X,Y,Z	坐标值参数
rstride	数组行距（步长大小）
cstride	数组列距（步长大小）
color	所有曲面块颜色
cmap	所有曲面块颜色映射
facecolors	单独曲面块表面颜色
vmin	映射最小值
vmax	映射最大值

其中cmap的渐变色参数取值见：
https://matplotlib.org/2.0.2/examples/color/colormaps_reference.html

代码如下：

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

fig = plt.figure()
ax = Axes3D(fig)
x=np.arange(-30,30,1)
y=np.arange(-30,30,1)
x, y = np.meshgrid(x, y)
z=x**2+y**2

ax.set_xlabel('X')
ax.set_ylabel('Y')
ax.set_zlabel('Z')

ax.plot_surface(x, y, z, cmap='rainbow')
plt.show()

散点图绘制

散点图绘制需要用到ax.scatter，具体代码如下

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

fig = plt.figure()
ax = Axes3D(fig)
x=np.arange(-30,30,1)
y=np.arange(-30,30,1)
x, y = np.meshgrid(x, y)
z=x**2+y**2

ax.set_xlabel('X')
ax.set_ylabel('Y')
ax.set_zlabel('Z')

ax.scatter(x, y, z,marker=".",color="blue")
plt.show()

更多随机散点图相关内容可见：https://blog.csdn.net/qq_43605229/article/details/115439364