python - 散点图图例仅显示一个带颜色的变量 -Pandas Seaborn

Question

我正在尝试使用 seaborn 包在 pandas 中绘制散点图。我希望我的两个变量都显示在图例中，但我只得到一个。以下是我所做的:

import pandas as pd
import numpy as np
from matplotlib import pyplot as plt
import seaborn as sns 

df = pd.DataFrame(np.random.randn(100, 6), columns=['a', 'b', 'c', 'd', 'e', 'f'])

我是这样绘制的，

plt.scatter(df['a'],df['b'], color = ['red', 'blue'])
plt.legend(loc = 'best')
plt.show()

我得到这样的图像，

如您所见，我没有看到蓝色的列/对象。我在这里犯了什么错误？我对此进行了搜索，但还没有运气。任何建议将不胜感激。

Answer 1

更新:

plt.scatter(df.index, df.a, color='red')
plt.scatter(df.index, df.b, color='blue')
plt.legend(loc = 'best')

结果:

<小时/>

我觉得你很困惑。

演示:

假设您的 DF 中只有三行:

In [53]: df = pd.DataFrame({'x':[1,2,3], 'y':[5,6,7]})

In [55]: df
Out[55]:
   x  y
0  1  5
1  2  6
2  3  7

您正在尝试绘制以下三点 - 您期望什么样的图例？

In [54]: plt.scatter(df.x, df.y, color=['red','blue'])
Out[54]: <matplotlib.collections.PathCollection at 0x149f9f28>

In [56]: plt.legend(loc = 'best')
Out[56]: <matplotlib.legend.Legend at 0x1353ca20>