matplotlib - 带有标量数据的散点图

Question

我想用 matplotlib 创建一个散点图，其中数据点附加了标量数据，并根据它们的附加值相对于集合中其他点的大小分配了一种颜色。即，我想要类似于热图的东西。但是，我正在寻找一个“离散”热图，即在原始数据集中没有点的地方不应该绘制任何东西，特别是不应该执行插值(在空间中)。

这能做到吗？

Answer 1

您可以使用分散，并将附加值设置为 c 参数:

import numpy as np
import pylab as pl

x = np.random.uniform(-1, 1, 1000)
y = np.random.uniform(-1, 1, 1000)

z = np.sqrt(x*x+y*y)

pl.scatter(x, y, c=z)
pl.colorbar()
pl.show()