如何使用Python pcolor解决尺寸问题？

Question

我有3个清单，我想作一个pcolor图。

ccplot = plt.pcolor（a，b，c，vmin = np.min（c），vmax = np.max（c））

a和b的形状为：（108，）

问题是：

当c的形状为（216，）时出现错误：

“ ValueError：没有足够的值可解包（预期2，得到1）”

当我将c重塑为形状为（10800，2）的2d数组时，出现错误：

“ TypeError：C（108，2）的尺寸与X（108）和/或Y（108）不兼容；请参阅help（pcolor）”

请帮助我处理这个pcolor图。 我预先感谢。

问候特拉维斯

Answer 1

我相信您正在错误地看待pcolor实现。 如果您将c作为（216，）的数组，则没有任何意义，因为它需要i和j值来解包才能在2D网格中找到其位置。 pcolor需要能够访问c [i，j]，对于像（216，）这样的一维数组，这是不可能的。

同样，当您有C（108,2）时，其他数组应看起来像X（109，）Y（3，）。 请尝试以下示例：

import matplotlib.pyplot as plt
import numpy as np
from matplotlib.colors import LogNorm

C = np.random.rand(6, 10)
X = range(11)
Y = range(7)

fig, ax0 = plt.subplots(1, 1)

c = ax0.pcolor(X, Y, C)
ax0.set_title('default: no edges')

plt.show()

因此c是一个6x10矩阵，需要x和y的拐角位置。 然后，它们的长度必须为7，长度为11，以提供c矩阵角的所有坐标。 对于角落的一些视觉效果，您可以转到https://matplotlib.org/api/_as_gen/matplotlib.pyplot.pcolor.html ，其中第一幅图像向您展示X，Y和C的关系。

希望能帮助到你！

Answer 2

问题在于阵列的尺寸必须兼容。 可以通过将a和b“重塑”为（36,3），然后通过ax.pcolormesh（）制作那些网格，从而实现该解决方案。 最后，如果c也重塑为（36,3），即转换为2D数组，则可以绘制pcolor。 请特别注意网格，以使其根据应用程序有意义。

a = a.reshape(36,3)
b = b.reshape(36,3)
c = c.reshape(36,3)

#making the mesh
a,b = np.meshgrid(a,b)

#pcolormesh
fig, ax0 = plt.subplots(1, 1)
c = ax0.pcolor(a, b, c)
plt.show()

pcolormesh更适合较大的数据集。