根据数值型变量设置图片的颜色

最新推荐文章于 2024-03-13 19:56:04 发布

原创最新推荐文章于 2024-03-13 19:56:04 发布 · 464 阅读

CC 4.0 BY-SA版权

%matplotlib notebook
from mpl_toolkits.mplot3d import Axes3D

label_color_dict = {1: 'red', 2: 'green', 3: 'blue', 4: 'yellow'}
colors = list(map(lambda label: label_color_dict[label], y_train))  #利用自己的来开始设置颜色

fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
ax.scatter(X_train['width'], X_train['height'], X_train['color_score'], c=colors, marker='o', s=100)
ax.set_xlabel('width')
ax.set_ylabel('height')
ax.set_zlabel('color_score')
plt.show()