python seaborn图中的图例标签不正确

时间:2017-07-28 15:55:27

标签: python matplotlib seaborn

enter image description here

上面的情节是用python中的seaborn制作的。但是,不确定为什么有些传奇圈子用颜色填充而其他圈子没有。这是我正在使用的色彩映射:

sns.color_palette("Set2", 10)

g = sns.factorplot(x='month', y='vae_factor', hue='ad_name', col='crop', data=df_sub_panel,
                   col_wrap=3, size=5, lw=0.5, ci=None, capsize=.2, palette=sns.color_palette("Set2", 10),
                   sharex=False, aspect=.9, legend_out=False)
g.axes[0].legend(fancybox=None)

- 编辑:

圆圈可以填充吗?他们没有填补的原因是他们可能没有这个特定情节中的数据

1 个答案:

答案 0 :(得分:6)

当没有数据时,圈子没有填写,我认为你已经推断过了。但是可以通过操纵图例对象来强制它。

完整示例:

import pandas as pd
import seaborn as sns

df_sub_panel = pd.DataFrame([
  {'month':'jan', 'vae_factor':50, 'ad_name':'China', 'crop':False},
  {'month':'feb', 'vae_factor':60, 'ad_name':'China', 'crop':False},
  {'month':'feb', 'vae_factor':None, 'ad_name':'Mexico', 'crop':False},
])

sns.color_palette("Set2", 10)

g = sns.factorplot(x='month', y='vae_factor', hue='ad_name', col='crop', data=df_sub_panel,
                   col_wrap=3, size=5, lw=0.5, ci=None, capsize=.2, palette=sns.color_palette("Set2", 10),
                   sharex=False, aspect=.9, legend_out=False)

# fill in empty legend handles (handles are empty when vae_factor is NaN)
for handle in g.axes[0].get_legend_handles_labels()[0]:
  if not handle.get_facecolors().any():
    handle.set_facecolor(handle.get_edgecolors())

legend = g.axes[0].legend(fancybox=None)

sns.plt.show()

重要的部分是在handle末尾(在for循环中)操纵legend个对象。

这将产生:

enter image description here

与原始版本(没有for循环)相比:

enter image description here

编辑:由于评论的建议,现在不那么狡猾了!