python - 带有来自单独 2D 数组的值的彩色 2D 网格

Question

我有两个数据数组，x 和 y。我想绘制 y 与 x 的散点图。 x 的范围为 [0,3]，y 的范围为 [-3, 3]。然后，我想将该区域网格化为 n × m 网格，并根据单独的 2D numpy 数组(与网格形状相同，n × m)的值对每个区域中的点进行着色。因此，我的绘图最左上角的网格单元应根据 colorarr[0][0] 等的值进行着色。有人对如何执行此操作有任何建议吗？到目前为止我发现的最接近的是:

2D grid data visualization in Python

不幸的是，这仅显示了 colorarr，而不是我想要可视化的 2D 区域。

谢谢!

Answer 1

我认为你想要的是二维直方图。 Matplotlib.pyplot让这变得非常容易。

import numpy as np
import matplotlib.pyplot as plt


# Make some points
npoints = 500
x = np.random.uniform(low=0, high=3, size=npoints)
y = np.random.uniform(low=-3, high=3, size=npoints)

# Make the plot
plt.hist2d(x, y)
plt.colorbar()
plt.show()