import matplotlib.pyplot as plt
%matplotlib notebook
from mpl_toolkits.mplot3d import Axes3D
fig = plt.figure()
# 创建一个3D坐标轴
ax = fig.add_subplot(111, projection='3d')
# ax.scatter(x, y, z)
df_1 = pd.DataFrame()
df_1['dim0'] = X_test[:,0]
df_1['dim1'] = X_test[:,1]
df_1['dim2'] = X_test[:,2]
df_1['labels'] = labels_pred
colors = ['red', 'purple', 'orange', 'green', 'blue']
for label in set(labels_pred):
tmp_df = df_1[df_1['labels'] == label]
tmp_labels = tmp_df['labels']
ax.scatter(tmp_df['dim0'], tmp_df['dim1'], tmp_df['dim2'],label="{}".format(label), s=1,c=colors[label],alpha=0.8)
ax.legend()
# 设置坐标轴标签
ax.set_xlabel('X Axis')
ax.set_ylabel('Y Axis')
ax.set_zlabel('Z Axis')
# 显示图像
plt.show()
pyplot画3D散点图并根据标签设置颜色
©著作权归作者所有,转载或内容合作请联系作者
- 文/潘晓璐 我一进店门，熙熙楼的掌柜王于贵愁眉苦脸地迎上来，“玉大人，你说我怎么就摊上这事。” “怎么了？”我有些...
- 文/花漫 我一把揭开白布。 她就那样静静地躺着，像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上，一...
- 文/苍兰香墨 我猛地睁开眼，长吁一口气：“原来是场噩梦啊……” “哼！你这毒妇竟也来了？” 一声冷哼从身侧响起，我...
推荐阅读更多精彩内容
- 这里有两种方式可供选择使用 第一种方式是使用visualMap视觉映射组件 第二种方式是通过itemStyle的c...
- 导读 主成分分析(Principal Components Analysis，PCA)，也称主分量分析或主成分回归...
