我正在尝试使用scatter
包在pandas
中绘制seaborn
图。我想让我的两个变量都显示在图例中,但我只得到一个。以下是我的所作所为:
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()
如您所见,我没有看到a
列/对象为蓝色。我在这里犯了什么错误?我搜索了这个,没有运气。任何建议将不胜感激。
答案 0 :(得分:2)
<强>更新强>
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>