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

2024-09-27 19:27:55 发布

您现在位置:Python中文网/ 问答频道 /正文

我有一个形状为(5,6)的numpy数组。我无法用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()

Tags: inimportnumpyforasnppltrandom

热门问题