如何使用colormap为matplotlib散点图中的特定点设置标记类型

Question

我有一个用户案例，假设我有三个系列数据：x，y，z。 我想使用（x，y）作为坐标，使用plt.scatter的cmap关键字作为散点的颜色制作散点图。 但是，我想通过使用与其他点不同的标记类型和大小来突出某些特定点。

最小的例子如下：

x,y,z = np.random.randn(3,10)
plt.scatter(x,y,c=z,cmap=matplotlib.cm.jet)
plt.colorbar()​

如果我想为（x [5]，y [5]，z [5]）使用不同的标记类型，我该怎么做？ 我能想到的唯一方法是使用plt.scatter（[x [5]，y [5]）再次绘制此点，但通过手动查找与z [5]对应的colormap颜色来定义颜色。 然而，这非常繁琐。 有没有更好的办法？

Answer 1

每个散点图都有一个标记，默认情况下，您不能在单个散点图中使用不同的标记。 因此，如果您乐意只更改markerize并使标记保持不变，则可以为scatter s参数提供不同大小的数组。

import matplotlib.pyplot as plt
import numpy as np; np.random.seed(10)

x,y,z = np.random.randn(3,10)

sizes = [36]*len(x)
sizes[5] = 121
plt.scatter(x,y,c=z,s=sizes, cmap=plt.cm.jet)

plt.colorbar()

plt.show()

如果您确实需要不同的标记样式，则可以绘制新的散点图。 然后，您可以将第二个散点图的颜色限制设置为第一个散点图的颜色限制。

import matplotlib.pyplot as plt
import numpy as np; np.random.seed(10)

x,y,z = np.random.randn(3,10)
xs, ys, zs = [x[5]], [y[5]], [z[5]]
print xs, ys, zs
y[5] = np.nan

sc = plt.scatter(x,y,c=z,s=36, cmap=plt.cm.jet)
climx, climy = sc.get_clim()
plt.scatter(xs,ys,c=zs,s=121, marker="s", cmap=plt.cm.jet, vmin=climx, vmax=climy  )  

plt.colorbar()

plt.show()

最后，在这个答案中将给出在同一散点图中有几个不同标记的复杂解决方案。