我正在尝试将图例正确显示在下图中。它应该只是“绿色”、“蓝色”和“红色”以及相应的颜色。但它无处不在。
代码如下:
import pandas as pd
import matplotlib.pyplot as plt
df = pd.DataFrame({
'category':['blue','green','red','blue','green','red','blue','green','red'],
'attempts':[8955,7881,6723,100,200,300,4567,876,54],
'success':[3000,7500,2000, 256,4567,4567,7665,543,43]
})
fig,ax = plt.subplots()
plt.scatter(df['attempts'],df['success'],c=df['category'],label=df['category'])
plt.legend(loc=2)
plt.savefig('scatter.png')
plt.show()
我怎样才能做到这一点? (这里有一个类似的:第二部分“带组的散点图”中的https://pythonspot.com/matplotlib-scatterplot/,但这不是基于pandas dataframe)。
最佳答案
您可以使用seaborn的散点图
:
fig,ax = plt.subplots()
sns.scatterplot(data=df, hue='category', x='attempts', y='success')
plt.legend(loc=2)
plt.savefig('scatter.png')
plt.show()
输出:
或者纯matplotlib:
fig,ax = plt.subplots()
for k,d in df.groupby('category'):
ax.scatter(d['attempts'], d['success'], label=k)
plt.legend(loc=2)
plt.savefig('scatter.png')
plt.show()
输出:
关于python - 来自 pandas 数据帧的散点图中的 Matplotlib 图例,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/57999419/