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

【问题标题】：1D heatmap with double squares on y axisy 轴上带有双正方形的一维热图
【发布时间】：2020-05-30 06:13:46
【问题描述】：

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

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()

输出图：

如果你能注意到，如果我有 1 维数组，该图会在 Y 轴上生成双正方形而不是一个。我怎样才能只保留其中一个？

【问题讨论】：

标签： python matplotlib heatmap

【解决方案1】：

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

将它们关闭（x 轴和 y 轴）

plt.grid(False)

如果要选择性显示，可以使用

plt.grid(axis='x')

或

plt.grid(axis='y')

【讨论】：