pyplot畫3D散點(diǎn)圖并根據(jù)標(biāo)簽設(shè)置顏色

import matplotlib.pyplot as plt
%matplotlib notebook
from mpl_toolkits.mplot3d import Axes3D

fig = plt.figure()
 
# 創(chuàng)建一個(gè)3D坐標(biāo)軸
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()
 
# 設(shè)置坐標(biāo)軸標(biāo)簽
ax.set_xlabel('X Axis')
ax.set_ylabel('Y Axis')
ax.set_zlabel('Z Axis')

# 顯示圖像
plt.show()
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成，瀏覽時(shí)請(qǐng)結(jié)合常識(shí)與多方信息審慎甄別。
平臺(tái)聲明：文章內(nèi)容（如有圖片或視頻亦包括在內(nèi)）由作者上傳并發(fā)布，文章內(nèi)容僅代表作者本人觀點(diǎn)，簡(jiǎn)書系信息發(fā)布平臺(tái)，僅提供信息存儲(chǔ)服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容