gpt4 book ai didi

matplotlib - Matplotlib 中的叠加轮廓图

转载 作者:行者123 更新时间:2023-12-03 13:32:07 25 4
gpt4 key购买 nike

我需要比较两组的二维分布。

当我使用 matplotlib.pyplot.contourf并覆盖图,每个等高线图的背景颜色填充整个图空间。有没有办法让每个等高线图的最低等高线级别透明,以便更容易看到每个等高线的中心?

import numpy as np
import matplotlib.pyplot as plt
from matplotlib import cm
import scipy.stats as st

def make_cloud(x, y, std, n=100):
x = np.random.normal(x, std, n)
y = np.random.normal(y, std, n)
return np.array(zip(x, y))

def contour_cloud(x, y, cmap):
xmin, xmax = -4, 4
ymin, ymax = -4, 4

xx, yy = np.mgrid[xmin:xmax:100j, ymin:ymax:100j]
positions = np.vstack([xx.ravel(), yy.ravel()])
values = np.vstack([x, y])
kernel = st.gaussian_kde(values)
f = np.reshape(kernel(positions).T, xx.shape)

plt.contourf(xx, yy, f, cmap=cmap, alpha=0.5)


cloud1 = make_cloud(-1, 1, 1)
cloud2 = make_cloud(1, -1, 1)

plt.scatter(x=cloud1[:,0], y=cloud1[:,1])
plt.scatter(x=cloud2[:,0], y=cloud2[:,1], color='red')

fig = plt.gcf()
ax = plt.gca()

contour_cloud(x=cloud1[:, 0], y=cloud1[:, 1], cmap=cm.Blues)
contour_cloud(x=cloud2[:, 0], y=cloud2[:, 1], cmap=cm.Reds)

enter image description here

最佳答案

您需要查看一些控件 contourf .您可以手动更改不同的级别,也可以更改高于/低于规范的颜色图。默认情况下,最低级别(或高于最大值)的区域的填充似乎是透明的。

所以,做你想做的最简单的方法是手动指定级别并指定它们,以便有 点低于最低水平,但 不是 任何高于最高级别的点。

如果更换:

plt.contourf(xx, yy, f, cmap=cmap, alpha=0.5)

和:
step = 0.02
m = np.amax(f)
levels = np.arange(0.0, m, step) + step
plt.contourf(xx, yy, f, levels, cmap=cmap, alpha=0.5)

产生如下图像:
enter image description here

有关高于/低于颜色图值的值行为的更多详细信息,请参阅 here .

关于matplotlib - Matplotlib 中的叠加轮廓图,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/42602425/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com