我正在尝试在同一张图像上绘制两个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")
答案 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)