图例仅显示一个元素

Question

我有以下数据：

   a  b  col lab
0  1  6    1   a
1  4  5    2   b
2  1  7    3   c
3  5  5    4   d
4  6  2    5   e

我的目标是绘制[a]和[b]的散点图，并使col的每个值使用不同的颜色，并有一个图例将颜色映射到标签上。

我尝试了以下操作，但只显示了第一个元素的图例。

plt.scatter(df['a'], df['b'], c = df['col'])
plt.legend(df['lab'])

我在做什么错了？