在seaborn情节中使用sns.set

时间:2015-07-02 17:32:44

标签: python seaborn

我已经找到了一个明确的答案并且找不到一个,如果之前已经问到这个我道歉。我使用seaborn 0.6和matplotlib 1.4.3。我想在ipython笔记本中创建许多数字时暂时改变绘图的样式。

具体来说,在这个例子中,我想基于每个图改变字体大小和背景样式。

这会创建我正在寻找的图,但全局定义参数:

import seaborn as sns
import numpy as np

x = np.random.normal(size=100)

sns.set(style="whitegrid", font_scale=1.5)
sns.kdeplot(x, shade=True);

然而这失败了:

with sns.set(style="whitegrid", font_scale=1.5):
    sns.kdeplot(x, shade=True);

使用:

---------------------------------------------------------------------------
AttributeError                            Traceback (most recent call last)
<ipython-input-10-70c5b03f9aa8> in <module>()
----> 1 with sns.set(style="whitegrid", font_scale=1.5):
      2     sns.kdeplot(x, shade=True);

AttributeError: __exit__

我也尝试过:

with sns.axes_style(style="whitegrid", rc={'font.size':10}):
    sns.kdeplot(x, shade=True);

哪个不会失败,但它也不会改变字体的大小。任何帮助将不胜感激。

2 个答案:

答案 0 :(得分:9)

最好的办法是将seaborn样式和上下文参数合并到一个字典中,然后将其传递给plt.rc_context函数:

import numpy as np
import seaborn as sns
import matplotlib.pyplot as plt  
x = np.random.normal(size=100)
with plt.rc_context(dict(sns.axes_style("whitegrid"),
                         **sns.plotting_context("notebook", font_scale=1.5))):
    sns.kdeplot(x, shade=True)

答案 1 :(得分:3)

这就是我正在使用的,利用matplotlib提供的上下文管理:

import matplotlib

class Stylish(matplotlib.rc_context):
    def __init__(self, **kwargs):
        matplotlib.rc_context.__init__(self)
        sns.set(**kwargs)

然后例如:

with Stylish(font_scale=2):
    sns.kdeplot(x, shade=True)