python - y 轴上有双方 block 的一维热图

Question

我使用此代码生成一维热图:

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 轴上生成两个正方形，而不是一个。我怎样才能只保留其中一个？

Answer 1

您看到的白线是网格线。您可以使用

将它们关闭(x 轴和 y 轴)

plt.grid(False)

如果你想有选择地显示，可以使用

plt.grid(axis='x')

或

plt.grid(axis='y')