python - PatchCollection 的 Matplotlib 颜色条覆盖颜色

Question

我正在根据变量值中设置的值绘制补丁。

pc = PatchCollection(patches, match_original=True)  

norm = Normalize()
cmap = plt.get_cmap('Blues')
pc.set_facecolor(cmap(norm(values)))
ax.add_collection(pc)

现在我还想要一个额外的颜色条。如果我插话(例如在 set_facecolor 之前)

pc.set_array(values) # values
plt.colorbar(pc)

它有效，但现在所有颜色都变成了灰度。以下 set_facecolor 不会改变任何东西。即使我在 plt.colorbar() 中添加了 cmap= 命令，所有内容仍保持灰度。我不知道该怎么办。

Answer 1

您是在问如何通过特定的颜色图为集合着色吗？

只需设置颜色图 (cmap) 以及 array。

例如:

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

num = 10
data = np.random.random((num, 2))
values = np.random.normal(10, 5, num)

# I'm too lazy to draw irregular polygons, so I'll use circles
circles = [plt.Circle([x, y], 0.1) for (x,y) in data]
col = PatchCollection(circles)

col.set(array=values, cmap='jet')

fig, ax = plt.subplots()

ax.add_collection(col)
ax.autoscale()
ax.axis('equal')

fig.colorbar(col)
plt.show()

解释你的问题是怎么回事:

对于集合，面部颜色要么手动设置(即 set_facecolors)，要么将它们颜色映射到由颜色条控制的一系列值(set_array)。这两个选项是相互排斥的。因此，在您调用 set_array 后，原始面部颜色将被忽略。