如何使用 3 列创建散点图 plot

Question

我有一个 df 看起来：

df_results = pd.DataFrame(data={})
df_results['names'] = ['James', 'Lucas', 'Henry', 'James', 'Lucas', 'Henry']
df_results['try_name'] = ["try_1", "try_1", "try_1", "try_2", "try_2", "try_2"]
df_results['score'] = [0.7, 0.9, 0.3, 0.91, 0.1, 0.2]

   names try_name  score
0  James    try_1   0.70
1  Lucas    try_1   0.90
2  Henry    try_1   0.30
3  James    try_2   0.91
4  Lucas    try_2   0.10

我想创建一个 plot （分散），其中：

• x 轴是列try_name
• y 轴是列score
• 值（图中的点）根据列names （3 种不同颜色）

如何创建这个 plot？

Answer 1

纯matplotlib解决方案：

fig = plt.figure(figsize=(7,4))
ax = fig.add_subplot(111)
for name in df_results.names.unique():
    ax.scatter(df_results[df_results.names == name].try_name, 
               df_results[df_results.names == name].score, label=name)
ax.legend()

使用 pandas plot：

fig = plt.figure(figsize=(7,4))
ax = fig.add_subplot(111)
for i,name in enumerate(df_results.names.unique()):
    df_results[df_results.names == name].plot.scatter('try_name', 'score', 
                                                      ax=ax, color='C{}'.format(i), 
                                                      label=name)
ax.legend()

Answer 2

您可以使用 seaborn 的swarmplot() 、 stripplot()或 scatterplot( scatterplot() 。

例如swarmplot() ：

import seaborn as sns
sns.swarmplot(data=df_results, x='try_name', y='score', hue='names')