二维数组matplotlib的颜色图

Question

所以，我觉得这会很简单，但我一直很难在一个可理解的例子中找到我想要的东西。

基本上我想绘制相位图，所以假设我有一个二维数组，我如何让matplotlib将其转换成一个可以附加标题、轴和图例（颜色条）的图。

我正在寻找一个非常简单的裸骨骼解决方案，它只使用将与任何二维阵列一起工作的所需内容。

我确定这很简单，我只是有点厚，但我真的有很多麻烦。

我一直在处理这些示例，但它们似乎不太适合我要做的工作：我喜欢这个图的一般外观，我只希望能够传入一个2dArray并得到相同的结果：

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

H = [[1,2,3,4][5,6,7,8][9,10,11,12][13,14,15,16]]

fig = plt.figure(figsize=(6, 3.2))

ax = fig.add_subplot(111)
ax.set_title('colorMap')
X,Y = np.meshgrid(xedges, yedges)
plt.pcolormesh(X, Y, H)
ax.set_aspect('equal')

cax = fig.add_axes([0.12, 0.1, 0.78, 0.8])
cax.get_xaxis().set_visible(False)
cax.get_yaxis().set_visible(False)
cax.patch.set_alpha(0)
cax.set_frame_on(False)
plt.colorbar(orientation='vertical')
plt.show()

Answer 1

下面是一个最简单的示例，其中包含关键的代码行：

import numpy as np 
import matplotlib.pyplot as plt

H = np.array([[1, 2, 3, 4],
          [5, 6, 7, 8],
          [9, 10, 11, 12],
          [13, 14, 15, 16]])

plt.imshow(H)
plt.show()

Answer 2

恐怕你的示例不起作用，因为X和Y没有定义。因此，与其使用pcolormesh，不如使用imshow：

import numpy as np
import matplotlib.pyplot as plt

H = np.array([[1, 2, 3, 4],
              [5, 6, 7, 8],
              [9, 10, 11, 12],
              [13, 14, 15, 16]])  # added some commas and array creation code

fig = plt.figure(figsize=(6, 3.2))

ax = fig.add_subplot(111)
ax.set_title('colorMap')
plt.imshow(H)
ax.set_aspect('equal')

cax = fig.add_axes([0.12, 0.1, 0.78, 0.8])
cax.get_xaxis().set_visible(False)
cax.get_yaxis().set_visible(False)
cax.patch.set_alpha(0)
cax.set_frame_on(False)
plt.colorbar(orientation='vertical')
plt.show()