python - 显示出现频率的标记颜色

Question

我有100个数据点，每个数据点都有一个时间序列。我计算了每对点之间的距离(代码中的dist)以及相应时间序列之间的相关系数(代码中的corr)。现在我需要有一个距离(在 x 轴)v/s 相关系数(在 y 轴)的散点图，并且标记颜色应该给出编号。每个距离值处相关系数的出现次数。我使用 matplotlib 尝试了以下代码

colors=np.random.randint(len(dist))
cmap=plt.cm.viridis
plt.scatter(dist,corr,c=colors,cmap=cmap)
plt.colorbar()
plt.show()

结果不正确。

使用散点图是否可以得到想要的结果？或者有其他方法可以获取吗？

Answer 1

您正在尝试生成一个带有单个数字作为颜色区分器的颜色图，即 len(colors) = 1，但您需要 len(colors) = len(dist)。

尝试:

colors=np.random.randint(len(dist), size=len(dist))

不确定您想要实现什么目标。也许这会起作用:

plt.scatter(dist,corr,c=dist,cmap=cmap)