y轴上有双正方形的一维热图

我用这个代码来生成一个1D热图:

from matplotlib.pyplot import figure
import matplotlib.pyplot as plt
import matplotlib as mpl
mpl.style.use('seaborn')

data = np.array([0.8373871, 0.48286343, 0.11166477, 0., 0.07306099, 1.,
                     0.8604202, 0.59167194, 0.60088444, 0.32089615]).reshape(1, 10)
plt.imshow(data, cmap='Reds')
plt.xlabel('Timesteps', weight='bold', fontsize=14)
plt.show()

Y