python使用matplotlib绘制三维空间散点图

Posted 2021-03-01 厚土

 1 # -*-coding: utf-8- -*-
 2 \'\'\'
 3 Create on 2020/04/14
 4 @author: Atwood Zhang
 5 \'\'\'
 6 
 7 import numpy as np
 8 import matplotlib.pyplot as plt
 9 import mpl_toolkits.mplot3d
10 
11 x = np.array([1, 2, 4, 5, 6])
12 y = np.array([2, 3, 4, 5, 6])
13 z = np.array([1, 2, 4, 5, 6])
14 
15 ax = plt.subplot(projection = \'3d\')  # 创建一个三维的绘图工程
16 ax.set_title(\'3d_image_show\')  # 设置本图名称
17 ax.scatter(x, y, z, c = \'r\')   # 绘制数据点 c: \'r\'红色，\'y\'黄色，等颜色
18 
19 ax.set_xlabel(\'X\')  # 设置x坐标轴
20 ax.set_ylabel(\'Y\')  # 设置y坐标轴
21 ax.set_zlabel(\'Z\')  # 设置z坐标轴
22 
23 plt.show()