Построение тепловой карты с помощью Python

Question

Я хочу сгенерировать тепловую карту, используя Python. Карта должна быть такой:

У меня есть массив с размером (n, n), и каждая "ячейка" содержит определенное значение. Чем выше это значение, тем больше должен быть розовый квадрат. Как я могу построить диаграмму такого типа, используя matplotlib? Есть ли другие библиотеки, которые я могу использовать?

Спасибо.

Answer 1

Вы можете попробовать это

n = 8
x = np.arange(n)
y = np.arange(n)
X, Y = np.meshgrid(x, y)
Z = np.random.randint(0, 800, (len(x), len(y)))
plt.figure()
plt.axes(aspect='equal')
plt.scatter(X+.5, Y+.5, Z, 'pink', marker='s')
plt.grid()
plt.xlim(0, n)
plt.ylim(0, n)
plt.tick_params(labelsize=0, length=0)