具有相同颜色条的多个seaborn kdeplot图

时间:2018-10-16 16:18:41

标签: python seaborn colorbar

我正在尝试在同一张图像上绘制两个kde分布,我想知道是否有一种方法可以对两个分布使用相同的“颜色范围”。

如果运行以下代码,您会看到最暗的红色和最暗的绿色对应于两个不同的浓度(0.04和0.15),而我希望两个色标都具有相同的范围,因此您可以轻松比较两种分布(即我想让最暗的红色和最暗的绿色对应相同的密度值)。

谢谢

import numpy as np
import seaborn as sns
import pandas
import matplotlib.pyplot as plt
from matplotlib import rcParams

np.random.seed(10)
sns.set(color_codes=True)

rcParams['font.family'] = 'sans-serif'
rcParams['font.sans-serif'] = ['Arial']

plt.ioff()

f, ax = plt.subplots(figsize=(15, 15))
ax.tick_params(axis='both', which='major', labelsize=22)

mean, cov = [0, 2], [(2, 1), (.5, 1)]
x1, y1 = np.random.multivariate_normal(mean, cov, size=50).T

mean, cov = [5, 7], [(3, 2), (7, 1)]
x2, y2 = np.random.multivariate_normal(mean, cov, size=50).T

ax = sns.kdeplot(x1, y1, cmap="Reds",   shade=True, shade_lowest=False, alpha=0.66, legend=False, cbar=True)
ax = sns.kdeplot(x2, y2, cmap="Greens", shade=True, shade_lowest=False, alpha=0.66, legend=False, cbar=True)

plt.xlabel("foo", fontsize=22)
plt.ylabel("bar", fontsize=22)
plt.savefig("foo_vs_bar.png")

enter image description here

1 个答案:

答案 0 :(得分:2)

可能为时已晚,但是我遇到了同样的问题。您可以指定kwarg中的级别,例如

kwargs = {'levels': np.arange(0, 0.15, 0.01)}

并传递到sns.kdeplot():

ax = sns.kdeplot(x1, y1, cmap="Reds",   shade=True, shade_lowest=False,
             alpha=0.66, legend=False, cbar=True, **kwargs)