python - 使用 matplotlib 的 pcolormesh 时指定颜色

Question

使用 matplotlib，我希望能够使用 pcolormesh 指定准确的颜色。这是我试过的

import numpy as np
from matplotlib import pyplot as plt

在顶部导入

X = np.linspace(0,1,100)
Y = np.linspace(0,1,100)
X,Y = np.meshgrid(X,Y)
Z = (X**2 + Y**2) < 1.
Z = Z.astype(int)
Z += (X**2 + Y**2) < .5

设置一堆假数据。 Z 只是0、1 和2(这就像我真正的问题)。

plt.pcolormesh(X,Y,Z,color=[(1,1,0),(0,0,1),(1,0,1)])

使用 color 参数调用 pcolormesh 试图获得黄色、蓝色和洋红色的图，但没有成功。事实上，我得到了默认颜色!

我的问题是:如何调用 pcolormesh 使第一个区域为黄色，第二个区域为蓝色，第三个区域为洋红色？

Answer 1

一种方法是使用自定义颜色图:

import numpy as np
from matplotlib import pyplot as plt
from matplotlib import colors as c

X = np.linspace(0,1,100)
Y = np.linspace(0,1,100)
X,Y = np.meshgrid(X,Y)
Z = (X**2 + Y**2) < 1.
Z = Z.astype(int)
Z += (X**2 + Y**2) < .5

cMap = c.ListedColormap(['y','b','m'])

plt.pcolormesh(X,Y,Z,cmap=cMap)
plt.show()