Ana*_*sia 4 matplotlib seaborn
我试图用seaborn绘制条形图.样本数据:
x=[1,1000,1001]
y=[200,300,400]
cat=['first','second','third']
df = pd.DataFrame(dict(x=x, y=y,cat=cat))
Run Code Online (Sandbox Code Playgroud)
我用的时候:
sns.factorplot("x","y", data=df,kind="bar",palette="Blues",size=6,aspect=2,legend_out=False);
Run Code Online (Sandbox Code Playgroud)
产生的数字是
当我添加图例时
sns.factorplot("x","y", data=df,hue="cat",kind="bar",palette="Blues",size=6,aspect=2,legend_out=False);
Run Code Online (Sandbox Code Playgroud)
结果图看起来像这样
如您所见,条形图从值移开.我不知道如何获得与第一张图中相同的布局并添加图例.
我不一定与seaborn绑在一起,我喜欢调色板,但任何其他方法都适合我.唯一的要求是该图看起来像第一个并且有图例.
看起来这个问题出现在这里 - 来自docs searborn.factorplot
色调:字符串,可选
用于按颜色分割绘图的数据中的变量名称.在"kind ="条的情况下
”
,这也会影响x轴上的位置.
所以,既然seaborn使用matplotlib,你可以这样做:
import matplotlib.pyplot as plt
import numpy as np
import seaborn as sns
x=[1,1000,1001]
y=[200,300,400]
sns.set_context(rc={"figure.figsize": (8, 4)})
nd = np.arange(3)
width=0.8
plt.xticks(nd+width/2., ('1','1000','1001'))
plt.xlim(-0.15,3)
fig = plt.bar(nd, y, color=sns.color_palette("Blues",3))
plt.legend(fig, ['First','Second','Third'], loc = "upper left", title = "cat")
plt.show()
Run Code Online (Sandbox Code Playgroud)
添加了@ mwaskom的方法来获得三种sns颜色.