基于具有自定义单元格宽度和高度的 Numpy 2d 数组创建 NonUniformImage

Question

我正在尝试使用 2d Numpy 数组的 Matplotlib 制作“热图”，并找到了 NonUniformImage 选项here。下面是一个简单的单一版本。不幸的是，我无法在不更改 y 的情况下更改 x。 pyhton Promppts 的错误是：Axes don't match array shape

可重现的示例

import numpy as np
import matplotlib.pyplot as plt
from matplotlib.image import NonUniformImage
from matplotlib import cm

x = np.arange(6)
y = np.arange(6)
z = x[:, np.newaxis].dot(y[:, np.newaxis].transpose())

fig, ax = plt.subplots()
im = NonUniformImage(ax, interpolation='nearest', extent=(0, len(x)-1, 0, len(y)-1))
im.set_data(x, y, z)
ax.images.append(im)
ax.set_xlim(0, len(x)-1)
ax.set_ylim(0, len(y)-1)

我的目标是基于带有“自定义网格单元”的 Numpy 2d 数组绘制地图。这意味着我可以为 x 和 y 分配不规则系列（例如np.array([0.5,3,8,15,15.4])。我知道分配的 x 和 y 代表单元格的中心，所以我最不想解决的问题是该图是完全可见的（如果可能的话）。

如何制作具有不同约束（不是正方形）的图像，自定义宽度和高度取决于两个列表？

Answer 1

我想你在定义z 时犯了一个错误。可能你追求的是

x = np.arange(6)
y = np.arange(8)
z = x[np.newaxis,:] * y[:, np.newaxis]
print(z.shape)

使得形状为 (8,6)。