如何从numpy数组排列热图(pcolormesh),而不在数组中转置数据?

时间:2019-06-12 15:05:29

标签: python numpy matplotlib heatmap colorbar

我有一个形状为(5,6)的numpy数组。除非我转置矩阵(如在其他线程中讨论的那样),否则无法使用pcolormesh创建热图,但是这样做也会将图形上的数据转置,这不是我想要的。我不移调时产生的错误很奇怪:

“ C(5,6)的尺寸与X(5)和/或Y(6)不兼容;请参阅help(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()

0 个答案:

没有答案