我有一个形状为(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()