# 绘制散点图 fig_scatter = plt.Figure(figsize=(6, 6), dpi=75) ax_scatter = fig_scatter.add_subplot(111) ax_scatter.scatter(df["Flavanoids"], df["OD"], c=df["Wine"]) ax_scatter.set_xlabel("Flavanoids") ax_scatter.set_ylabel("OD") ax_scatter.set_title("Wine") scatter_canvas = FigureCanvasTkAgg(fig_scatter, master=plot_window) scatter_canvas.draw() scatter_canvas.get_tk_widget().pack(side="left", fill=tk.BOTH, expand=True)怎么给不同类的散点指定颜色
时间: 2024-03-10 13:47:43 浏览: 18
要给不同类的散点指定颜色,可以通过创建一个字典,将不同类别的取值映射到不同的颜色上,然后将 scatter() 方法中的 c 参数设置为对应的 Series 对象。
例如,假设 df["Wine"] 表示三种不同的葡萄酒类别,可以创建一个字典来指定每种类别的颜色:
```
color_dict = {1: "red", 2: "green", 3: "blue"}
```
然后,将 df["Wine"] 中的每个值都映射到对应的颜色上:
```
colors = df["Wine"].map(color_dict)
```
最后,在 scatter() 方法中指定 c 参数为 colors:
```
ax_scatter.scatter(df["Flavanoids"], df["OD"], c=colors)
```
这样就可以根据不同类别的数据自动设置不同的颜色了。