python - Matplotlib:使用颜色图并对不同值使用不同标记

Question

我目前在数组 xy 中存储了一堆 (x,y) 点，我使用第三个数组 Kmap 进行着色，使用 matplotlib 内置的 cmap选项。

plt.scatter(xy[:, 0], xy[:, 1], s=70, c=Kmap, cmap='bwr')

这很好。现在，我想做一些额外的事情。在继续使用cmap的同时，我想根据Kmap值是>0、<0还是=0来使用不同的标记。我该怎么做呢？

注意:我可以想象分解这些点，并使用 if 语句使用不同的标记分别对它们进行散点绘制。但是，我不知道如何对这些值应用连续的 cmap。

Answer 1

分离数据集看起来是一个不错的选择。为了保持颜色之间的一致性，您可以使用 scatter 方法的 vmin、vmax 参数

import matplotlib.pyplot as plt
import numpy as np


#generate random data
xy = np.random.randn(50, 2)
Kmax = np.random.randn(50)

#data range
vmin, vmax = min(Kmax), max(Kmax)

#split dataset
Ipos = Kmax >= 0. #positive data (boolean array)
Ineg = ~Ipos      #negative data (boolean array)


#plot the two dataset with different markers
plt.scatter(x = xy[Ipos, 0], y = xy[Ipos, 1], c = Kmax[Ipos], vmin = vmin, vmax = vmax, cmap = "bwr", marker = "s")
plt.scatter(x = xy[Ineg, 0], y = xy[Ineg, 1], c = Kmax[Ineg], vmin = vmin, vmax = vmax, cmap = "bwr", marker = "o")

plt.show()