matplotlib绘制三维散点图示例

Posted 2022-03-14 非晚非晚

下列代码中，展示了在同一个坐标系下，同时绘制两种不同颜色颜色的散点图

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

#生成随机数据
data1 = np.random.randint(30, 70, size=(30, 3)) #30*3维度的[30,70]的随机整数
data2 = np.random.randint(10, 30, size=(40, 3)) #40*3维度的[10，30]的随机整数

x1 = data1[:, 0]
y1 = data1[:, 1]
z1 = data1[:, 2]

x2 = data2[:, 0]
y2 = data2[:, 1]
z2 = data2[:, 2]


# 绘制散点图
fig = plt.figure()
ax = Axes3D(fig)

'''
marker：形状，默认圆球('^'倒三角形状。)
c:颜色，默认青色

'''
ax.scatter(x1, y1, z1, c='r', marker = "^", label='red points')
ax.scatter(x2, y2, z2, c='g', label='green points')

# 绘制图例
ax.legend(loc='best')

# 添加坐标轴
ax.set_zlabel('Z Label', fontdict='size': 15, 'color': 'red') 
ax.set_ylabel('Y Label', fontdict='size': 15, 'color': 'red')
ax.set_xlabel('X Label', fontdict='size': 15, 'color': 'red')
plt.show()

结果显示：