我在 python 中繪制了一個 3-D 圖，顯示了 3 個特征（比如 A、B、C），每個特征都由目標變數“y_train”著色，我想在圖中添加一個圖例以顯示每種顏色的相關類別，但它只顯示其中一個“2”任何想法如何解決這個問題？
from mpl_toolkits.mplot3d import Axes3D
fig = plt.figure(figsize=(4,4))
ax = fig.add_subplot(111, projection='3d')
ax.scatter(X_train["A"],X_train["B"],X_train["C"], c = y_train)
ax.set_xlabel("A")
ax.set_ylabel("B")
ax.set_zlabel("C")
ax.legend(y_train)
plt.show()
uj5u.com熱心網友回復：
對于自動圖例創建，該方法PathCollection.legend_elements將完成這項作業。您只需將其打開包裝并將其插入圖例即可。
from mpl_toolkits.mplot3d import Axes3D
fig = plt.figure(figsize=(4,4))
ax = fig.add_subplot(111, projection='3d')
scatter = ax.scatter(X_train["A"],X_train["B"],X_train["C"], c=y_train)
legend = ax.legend(*scatter.legend_elements())
ax.set_xlabel("A")
ax.set_ylabel("B")
ax.set_zlabel("C")
ax.legend(legend)
plt.show()
轉載請註明出處，本文鏈接：https://www.uj5u.com/net/457245.html
標籤：Python matplotlib 分类