分类级别使用不同颜色的散点图

使用 Matplotlib

使用 Matplotlib 创建散点图，其中不同的分类级别由不同颜色表示，按照以下步骤操作：

1. 导入 Matplotlib 和要绘制的数据框。
2. 定义一个字典，将分类级别映射到绘制颜色。
3. 使用plt.scatter，传入 x 和 y 值以及 c 参数来指定颜色。

import matplotlib.pyplot as plt
import pandas as pd

colors = {'D':'tab:blue', 'E':'tab:orange', 'F':'tab:green', 'G':'tab:red', 'H':'tab:purple', 'I':'tab:brown', 'J':'tab:pink'}

df.scatter(df['carat'], df['price'], c=df['color'].map(colors))

plt.show()

使用 Seaborn

Seaborn 是 Matplotlib 的包装器，提供了更加用户友好的界面。要使用 Seaborn 创建具有不同颜色的分类级别散点图，请按照以下步骤操作：

1. 导入 Seaborn 和要绘制的数据框。
2. 使用 seaborn.scatterplot，传递在x和y值以及hue参数中指定分类级别。

import seaborn as sns

sns.scatterplot(x='carat', y='price', data=df, hue='color')

plt.show()

使用 pandas.groupby & pandas.DataFrame.plot

您还可以使用 pandas.groupby 和 pandas.DataFrame.plot 创建具有不同颜色的分类散点图水平。此方法需要更多的手动工作，但它可以让您更好地控制绘图的外观。

1. 导入 pandas 和要绘制的数据框。
2. 按分类级别。
3. 遍历组并用不同的颜色绘制每个组。

import pandas as pd

fig, ax = plt.subplots(figsize=(6, 6))

grouped = df.groupby('color')
for key, group in grouped:
    group.plot(ax=ax, kind='scatter', x='carat', y='price', label=key, color=colors[key])

plt.show()