python - 在轴的范围内传播 imshow 热图

Question

我在一个时间步有一组 x,y 位置，但我知道随着时间的推移，这个范围会扩大（已经由图像范围设置）。有没有办法使范围内的其余网格为 0，直到它被填充。我的理解是，来自 np.histogram2d 的 x,y 位置或地图本身需要在不同大小的新网格上进行重组，尽管我不确定如何。到目前为止，我有：

heatmap, xedges, yedges = np.histogram2d(x,y, bins=50)
extent = [xedges[0], xedges[-1], yedges[0], yedges[-1]]
ax.imshow(heatmap.T, origin='lower',extent=extent,cmap='cubehelix')
ax.set_xlim([20,220])
ax.set_ylim([-1,1])

但这会导致受限区域。我想让空白基本上变成黑色，直到新的 x,y 位置在稍后的某个时间步填充它们。

Answer 1

要独立于传入的数据来控制直方图的范围numpy.histogram2d，请将 bin 位置指定为数组。

例如，类似：

import numpy as np
import matplotlib.pyplot as plt

# Known ranges for the histogram and plot
xmin, xmax = 20, 220
ymin, ymax = -1, 1

# Generate some random data
x = np.random.normal(48, 5, 100)
y = np.random.normal(0.4, 0.1, 100)

# Create arrays specifying the bin edges
nbins = 50
xbins = np.linspace(xmin, xmax, nbins)
ybins = np.linspace(ymin, ymax, nbins)

# Create the histogram using the specified bins
data, _, _ = np.histogram2d(x, y, bins=(xbins, ybins))

# Plot the result
fig, ax = plt.subplots()
ax.imshow(data.T, origin='lower', cmap='cubehelix', aspect='auto',
          interpolation='nearest', extent=[xmin, xmax, ymin, ymax])

ax.axis([xmin, xmax, ymin, ymax])
plt.show()