matplotlib 如何在seaborn kdeplut中添加颜色栏

jucafojl 于 2023-03-30 发布在其他

关注(0)|答案(2)|浏览(111)

我想用seaborn.kdeplot创建一个Kernel-Density-Estimation，旁边有一个颜色条。

import matplotlib.pyplot as plt
import seaborn as sns
import numpy as np; np.random.seed(10)
import seaborn as sns; sns.set(color_codes=True)

mean, cov = [0, 2], [(1, .5), (.5, 1)]
x, y = np.random.multivariate_normal(mean, cov, size=50).T
sns.kdeplot(x, y ,shade=True)
plt.show()

当创建了Kernel-Density-Estimation时，我不知道如何创建colorbar。我尝试使用plt.colorbar()没有成功。

matplotlib

来源：https://stackoverflow.com/questions/39328551/how-to-add-a-colorbar-to-a-seaborn-kdeplot

2条答案

按热度按时间

ndasle7k1#

**现已实现it's！**参数cbar=True。

也可以使用shade_lowest=False不着色第一个级别。

import seaborn as sns
import numpy as np
import matplotlib.pylab as plt

x, y = np.random.randn(2, 300)
sns.kdeplot(x=x, y=y, zorder=0, n_levels=6, shade=True, 
    cbar=True, shade_lowest=False, cmap='viridis')

赞(0）回复(0）举报 2023-03-30

kpbwa7wx2#

你必须直接调用scipy KDE和matplotlib contour函数，但这只是一些额外的代码：

import matplotlib.pyplot as plt
import seaborn as sns
import numpy as np; np.random.seed(10)
import seaborn as sns; sns.set(color_codes=True)
from scipy import stats

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

kde = stats.gaussian_kde(data)
xx, yy = np.mgrid[-3:3:.01, -1:4:.01]
density = kde(np.c_[xx.flat, yy.flat].T).reshape(xx.shape)

f, ax = plt.subplots()
cset = ax.contourf(xx, yy, density, cmap="viridis")
f.colorbar(cset)

赞(0）回复(0）举报 2023-03-30

我来回答

matplotlib 如何在seaborn kdeplut中添加颜色栏

2条答案

相关问题

热门标签

最新问答