1

我正在尝试绘制一个 2d 直方图,显示两次测量之间收到的错误代码之间的相关性。错误代码值范围从 -3 到 5。我希望直方图为每个错误代码显示一个条形图。在由一次测量收到的两个错误代码标记的字段中,条形应该增加(改变颜色)。我在这里画了一个简短的草图。在此处输入图像描述

到目前为止,我只有下面的代码,不幸的是它没有给我想要的情节。有人知道如何得到我上面描述的情节吗?

data1=np.random.randint(-3,5,100)
data2=np.random.randint(-3,5,100)
fig, ax = plt.subplots()
ax.hist2d(data1, data2, bins=10)
plt.Axes.set_xlim(ax,left=-3, right=6)
plt.Axes.set_ylim(ax, bottom=-3, top=6)
plt.grid(True)
plt.show()
4

1 回答 1

2

您需要正确的数量bins和轴限制才能获得所需的可视化效果。此外,当您使用 时data1=np.random.randint(-3,5,100),您得到的最大整数是 4 而不是 5。下面是您的代码的修改版本。

data1=np.random.randint(-3,6,100)
data2=np.random.randint(-3,6,100)
n_bins = len(set(data1.flatten())) - 1

l = -3
r = 5

fig, ax = plt.subplots()
im = ax.hist2d(data1, data2, bins=n_bins+1)
ax.set_xlim(left=l, right=r)
ax.set_ylim(bottom=l, top=r)

shift = (im[1][1]-im[1][0])/2
plt.xticks(im[1], range(l, r+1, 1))
plt.yticks(im[1], range(l, r+1, 1))
plt.grid(True)
plt.colorbar(im[3])
plt.show()

在此处输入图像描述

于 2019-01-14T10:49:52.670 回答