如何使用从数组读取的颜色绘制例如正方形的 xy 网格

Question

我有单独的 x 和 y 坐标数组，以及相应值的 z 数组。我希望制作一个在每个 x 和 y 坐标处都有正方形的图，这些正方形具有从 z 数组设置的颜色 - 类似于

Answer 1

谢谢安德鲁。我现在明白这是如何运作的了。问题是我的 z 数组只是一列数字。由于它们没有任何合理的顺序，因此很难将数组重新整形为 2D 来使用 pcolor。

我想出了一个更好的解决方案，使用 for 循环将矩形补丁附加到补丁集合中，然后将颜色图分配给整个集合和绘图。

import numpy as np
import matplotlib
import matplotlib.pyplot as plt
from matplotlib.collections import PatchCollection
import matplotlib.patches as mpatches

fig = plt.figure(figsize=(9,5))
ax = plt.axes([0.1,0.1,0.7,0.7])
cmap = matplotlib.cm.jet
patches = []

data=np.array([4.5,8.6,2.4,9.6,11.3])
data_id_nos=np.array([5,6,9,8,7])
x_coords=np.array([3.12,2.6,2.08,1.56,1.04])
y_coords=np.array([6.76,6.24,5.72,5.20,4.68])
coord_id_nos=np.array([7,9,6,5,8])    

for i in range(len(data_id_nos)):
        coords=(x_coords[np.where(coord_id_nos == data_id_nos[i])],y_coords[np.where(coord_id_nos == data_id_nos[i])])
        art = mpatches.Rectangle(coords,0.50,0.50,ec="none")
        patches.append(art)

#create collection of patches for IFU position
IFU1 = PatchCollection(patches, cmap=cmap)
#set the colours = data values
IFU1.set_array(np.array(data))
ax.add_collection(IFU1)
plt.axis('scaled')
plt.xlabel('x (arcsecs)')
plt.ylabel('y (arcsecs)')

Answer 2

我猜你想要 pcolor，如此处所示。

Answer 3

您需要使用这个精确的代码执行类似的操作，

x = np.arange(10)
y = np.arange(10)
z = np.zeros([10,10])
z[1,5] = 10
z[2,7] = 20
z[3,9] = 30
pcolor(x,y,z)

最后一个点将偏离轴，但它应该给您一个想法。