Matplotlib新手求助：如何用多标记、多颜色展示数据

嘿，作为Matplotlib新手遇到这种可视化需求很正常，我来帮你把代码补全并解释清楚怎么实现多标记+多颜色的散点展示！

首先看你的代码思路是对的——把数据分成不同组别，每组用不同标记，同时共享同一个颜色映射来统一颜色含义。下面是完善后的完整代码，我会一步步拆解说明：

import matplotlib.pyplot as plt

x = [1,2,3,4,5,6,7,8,9,10]
y = [1,2,3,4,5,6,7,8,9,10]
colors = [0,1,1,0,1,4,1,3,2,4]
sizes = [500,500,300,300,300,500,500,300,300,300]

# 创建包含5个离散颜色的jet色阶（对应你colors里的0-4数值）
map1 = plt.cm.get_cmap("jet", 5)

# 分三组绘制散点，每组指定不同marker
scatter1 = plt.scatter(x[0:3], y[0:3], c=colors[0:3], s=sizes[0:3], marker="*", cmap=map1)
scatter2 = plt.scatter(x[3:6], y[3:6], c=colors[3:6], s=sizes[3:6], marker="<", cmap=map1)
scatter3 = plt.scatter(x[6:10], y[6:10], c=colors[6:10], s=sizes[6:10], marker="D", cmap=map1)

# 添加颜色条，明确颜色对应的数值
plt.colorbar(scatter1, ticks=[0,1,2,3,4], label="Color Value")

# 自定义图例，区分不同标记的组别
from matplotlib.lines import Line2D
custom_legend_elements = [
    Line2D([0], [0], marker='*', color='w', markerfacecolor='gray', markersize=15, label='Group 1'),
    Line2D([0], [0], marker='<', color='w', markerfacecolor='gray', markersize=15, label='Group 2'),
    Line2D([0], [0], marker='D', color='w', markerfacecolor='gray', markersize=15, label='Group 3')
]
plt.legend(handles=custom_legend_elements, loc="upper left")

# 添加基础标注
plt.xlabel("X Axis")
plt.ylabel("Y Axis")
plt.title("Multi-Marker & Multi-Color Scatter Plot")
plt.show()

关键细节解释： #

• 统一颜色映射：所有scatter调用都用同一个cmap=map1，这样colors里的同一个数值（比如1）在所有组别里显示的颜色是一致的，不会出现颜色混乱。
• 离散颜色设置：plt.cm.get_cmap("jet",5)把jet色阶拆成了5个离散颜色，刚好匹配你的colors里0-4的数值范围，每个数值对应一个固定颜色。如果你的颜色值范围不同，可以用plt.Normalize(min(colors), max(colors))来归一化。
• 自定义图例：默认的散点图图例会只展示颜色，无法区分标记类型，所以我们用Line2D手动创建图例元素，每个元素对应一个标记，这样读者能清楚知道每个标记代表哪组数据。
• 颜色条优化：指定ticks=[0,1,2,3,4]让颜色条的刻度和你的颜色值完全对应，加上label说明颜色的含义，让可视化更清晰。

这样运行代码后，你就能得到一个既有不同标记区分组别，又有统一颜色映射展示数值的散点图啦！

内容的提问来源于stack exchange，提问作者Thanh Bui