FacetGrid 图例为空

FacetGrid Legend empty

我正在尝试创建一个 seaborn FacetGrid 来显示我拥有的某些数据中集群之间的转换概率。数据有一堆主题和 4 个集群(因此每个主题有 16 个数据点,每对新旧集群一个)。因为这些是概率,所有具有相同旧集群(对于每个主题)的值总和为 1。我想使用 FacetGrid 点图(或堆叠条形图,但 seaborn 似乎不可能)显示此数据,例如每个旧集群都是一个单独的列,新集群是一个单独的色调,每个方面都显示主题与概率。

这是生成与我的功能相同的玩具 DataFrame 的代码,以及我想要制作的 facetgrid 的代码。然而,当我做这一切时,传奇是空的。

df,index = [],0
for subj in [1,2]:
    for i,j in itertools.product(range(4),range(4)):
        df.append(pd.DataFrame(index=[index],data=dict(subject=subj,old=i,new=j,prob=[.23,.24,.26,.27][j])))
        index+=1
df = pd.concat(df)
g = sns.FacetGrid(df,col='old',hue='new',margin_titles=False,legend_out=True)
g.map(sns.pointplot,'subject','prob',join=False)
g.add_legend()

查看 g._legend_data 表明它是空的,每个轴 get_legend_handles_labels() 都没有 returns 功能。怎么回事?

我无法 post 我的情节的图像,因为我没有 10 个声望;图例的名字出现在最右边,但里面什么也没有...

编辑:另外,我想稍微躲避这些值,但是 g.map(sns.pointplot,'subject','prob',join=False,dodge=.1) 看起来与上面完全相同,对于各种不同的躲避值...

编辑 2: 所以看起来使用 sns.factorplot 而不是 FacetGrid,图例和闪避都有效。我唯一不能使用 sns.factorplot 做的事情是旋转 xtick 标签,但无论如何在 Facet Grid 中这对 col_wrap 不起作用...为什么 FacetGrid 有这个问题?

正如您所注意到的,factorplot 提供了一个更容易同时使用 pointplotFacetGrid 的界面。图例和闪避不能按照您原来的方式工作的原因是在 FacetGrid 中设置 hue 和在 pointplot 中设置是不同的。当你在没有hue变量的情况下调用pointplot时,它不知道如何闪避,它认为它不需要添加任何图例数据。

选项一是移动到您使用 hue 变量的位置:

g = sns.FacetGrid(df,col='old', col_wrap=2)
g.map(sns.pointplot, 'subject', 'prob', 'new', dodge=.1, join=False)
g.add_legend()

但我想不出任何理由以这种方式使用 FacetGrid 而不是仅使用 factorplot:

g = sns.factorplot(x="subject", y="prob", hue="new", data=df,
               col="old", col_wrap=2, size=3,
               dodge=.1, join=False)

所以唯一剩下的问题是旋转 x 轴刻度标签。 factorplot returns 一个 FacetGrid 实例,所以原则上应该很容易使用进一步的 FacetGrid 方法来调整情节。我怀疑您遇到的问题是 this bug,它将在下一版本中修复。同时,您可以像这样解决它:

g = sns.factorplot(x="subject", y="prob", hue="new", data=df,
                   col="old", col_wrap=2, size=3,
                   dodge=.1, join=False)
for ax in g.axes:
    plt.setp(ax.get_xticklabels(), rotation=45)