如何在Seaborn的``山脊图''上将单位添加到轴标签

时间:2019-01-16 11:26:54

标签: python matplotlib seaborn

我想根据原始的“山脊图”示例(https://seaborn.pydata.org/examples/kde_ridgeplot.html)进行以下操作:

只需将单位添加到x轴的x标签上,以使其在图中看起来像 x in [m] 。我怎么做?将命令g.map(label, "x")扩展到g.map(label, "x in [m]")会导致以下错误: KeyError:“ ['x [m]']不在索引中”

我的代码如下:

import numpy as np
import pandas as pd
import seaborn as sns
import matplotlib.pyplot as plt
sns.set(style="white", rc={"axes.facecolor": (0, 0, 0, 0)})


errorNames = ['Error 1 - abc.....',
              'Error 2 - abc.....',
              'Error 3 - abc.....',
              'Error 4 - abc.....',
              'Error 5 - abc.....',
              'Error 6 - abc.....',
              'Error 7 - abc.....',
              'Error 8 - abc.....',
              'Error 9 - abc.....',
              'Error 10 - abc.....',
              'Error 11 - abc.....',
              'Error 12 - abc.....',
              'Error 13 - abc.....']


# Create the data
rs = np.random.RandomState(1979)
x = rs.randn(650)
#g = np.tile(list("ABCDEFGHIJKLM"), 50)
g = np.tile(list(errorNames), 50)

df = pd.DataFrame(dict(x=x, g=g))
#m = df.g.map(ord)
#df["x"] += m

# Initialize the FacetGrid object
pal = sns.cubehelix_palette(10, rot=-.25, light=.7)
g = sns.FacetGrid(df, row="g", hue="g", aspect=15, height=.5, palette=pal)

# Draw the densities in a few steps
g.map(sns.kdeplot, "x", clip_on=False, shade=True, alpha=1, lw=1.5, bw=.2)
g.map(sns.kdeplot, "x", clip_on=False, color="w", lw=2, bw=.2)
g.map(plt.axhline, y=0, lw=2, clip_on=False)


#sns.plt.xlim(-10, 3)


# Define and use a simple function to label the plot in axes coordinates
def label(x, color, label):
    ax = plt.gca()
    ax.text(0, .2, label, fontweight="bold", color=color,
            ha="left", va="center", transform=ax.transAxes)

g.map(label, "x")
#g.map(label, "x [m]")

# Set the subplots to overlap 
# Erst hier wird geplotted
g.fig.subplots_adjust(hspace=-.25)


# Remove axes details that don't play well with overlap
g.set_titles("")
g.set(yticks=[])
g.despine(bottom=True, left=True)

3 个答案:

答案 0 :(得分:1)

只是一些玩转轴对象g的尝试,我发现了一种使用set_xlabels的简单而看似直接的方法。感谢@DavidG查找文档here。具体来说,

  

set_xlabels([label])在网格的底部行上标记x轴。

g.map(label, "x")
g.set_xlabels('x in [m]')

enter image description here

答案 1 :(得分:0)

一种方法是使用g.axes获取轴的数组。然后获取数组中的最后一个轴并设置该轴的xlabel:

last_ax = g.axes.flat[-1]
last_ax.set_xlabel("x in [m]")

enter image description here

答案 2 :(得分:0)

还有一点需要注意,您可能希望将pal中的数量增加到13以保持颜色渐变。

pal = sns.cubehelix_palette(13, rot=-.25, light=.7)