1D тепловая карта с двойными квадратами по оси Y

Question

Я использую этот код для создания 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()

График вывода:

, если вы можете заметить, если хотя у меня есть 1 массив, график генерирует двойные квадраты по оси Y вместо одного. Как мне сохранить только один из них?

Answer 1

белые линии, которые вы видите, являются линиями сетки. Вы можете просто отключить их (как по оси x, так и по оси y), используя

plt.grid(False)

Если вы хотите показать выборочно, вы можете использовать

plt.grid(axis='x')

или

plt.grid(axis='y')