下列代码中,展示了在同一个坐标系下,同时绘制两种不同颜色颜色的散点图
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()
结果显示: