如何安排热图pcolormesh的数据？

Question

我一直在使用的所有例子都使用随机数据.我的问题是安排我的实际数据以适应解决方案.

我正在尝试创建x,y,z的热图/散点图:x和y是位置,而z是颜色.它们是三个相等长度的阵列.

X = [-0.11, -0.06, -0.07, -0.12, ...]
Y = [0.09, 0.13, 0.17, 0.09, ...]
Z = [0.38, 0.37, 0.44, 0.33, ...]

该pcolormesh文档似乎并没有描述什么"C"是只是说它可以是一个"蒙面阵列".可悲的是,我不知道那是什么(还).

如何将我的三个数组转换为它正在寻找的任何数组？我尝试将它们粘贴到一个numpy数组并传递它,这平息了没有"形状"的错误,但三维数组似乎不是它正在寻找的.

Answer 1

pcolormesh 想要三个2-D阵列.

你X和Y都很好,但需要通过numpy的的运行meshgrid功能,即:

import numpy as np
X = [-0.11, -0.06, -0.07, -0.12, ...]
Y = [0.09, 0.13, 0.17, 0.09, ...]
xx, yy = np.meshgrid(X, Y)

那么你只需要得到Z是在一个形状是一样的xx,并yy和你将所有设置.

对于散点图X,Y和,Z是罚款:

import matplotlib.pyplot as plt
X = [-0.11, -0.06, -0.07, -0.12, ...]
Y = [0.09, 0.13, 0.17, 0.09, ...]
Z = [0.38, 0.37, 0.44, 0.33, ...]
fig, ax = plt.subplots()
ax.scatter(X, Y, c=Z)
plt.show()