时间戳

首页 > 解决方案 > 如何创建具有 3 列的散点图

python - 如何创建具有 3 列的散点图

问题描述

我有一个 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

我想创建一个图（散点图），其中：

我怎样才能创建这个情节？

标签： pythonpython-3.xpandasmatplotlib

解决方案

纯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()

使用熊猫图：

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()

推荐阅读