Как выстроить тепловую карту (pcolormesh) из массива numpy без транспонирования данных в массиве? - PullRequest
Новая
       36

Как выстроить тепловую карту (pcolormesh) из массива numpy без транспонирования данных в массиве?

0 голосов
/ 12 июня 2019

У меня есть массив с формой (5,6).Я не могу создать тепловую карту с помощью pcolormesh, если не перенесу свою матрицу (как обсуждалось в других потоках), но это также транспонирует данные на графике, что не является тем, что я хочу.Ошибка, возникающая, когда я не транспонирую, странна:

"Размеры C (5, 6) несовместимы с X (5) и / или Y (6); см. Справку (pcolormesh)"

Мой код выглядит следующим образом (ScoresMatrix имеет размеры (5,6): 

import matplotlib.pyplot as plt
import numpy as np
import random

scoresMatrix = np.zeros((5,6))

i = 0
for row in scoresMatrix:
    j = 0
    for column in row:
        scoresMatrix[i,j] = random.randint(0,9)

x = np.arange(0, scoresMatrix.shape[0])
y = np.arange(0, scoresMatrix.shape[1])

X, Y = np.meshgrid(x, y)

plt.pcolormesh(X, Y, scoresMatrix, cmap = 'jet')
plt.colorbar()

plt.show()
...