你可以在不同的x位置绘制1D热图。下面的方法使用imshow来创建这样的热图。它假设每个时间步的强度存储在一个2D数组中。

import matplotlib.pyplot as plt
from matplotlib.cm import ScalarMappable
import numpy as np

num_steps = 20
num_intensities = 50
# first, create some random test data
data = np.random.randn(num_intensities, num_steps)
data = data.cumsum(axis=1).cumsum(axis=0)
data -= data.min(axis=1, keepdims=True)
data = data / data.max(axis=1, keepdims=True) * 7

fig, ax = plt.subplots()
cmap = plt.get_cmap('turbo')
norm = plt.Normalize(data.min(), data.max())
for step in range(num_steps):
    ax.imshow(data[:, step:step + 1], extent=[step - 0.3, step + 0.3, 0, 1],
              aspect='auto', origin='lower', cmap=cmap, norm=norm)

ax.set_xlim(-0.5, num_steps - 0.5)
ax.set_ylim(0, 1)
ax.set_xticks(range(num_steps))
ax.spines['top'].set_visible(False)
ax.spines['right'].set_visible(False)

plt.colorbar(ScalarMappable(cmap=cmap, norm=norm))
plt.tight_layout()
plt.show()