python - 根据 matplotlib 中线图中的密度(存储在数组中)更改图的颜色

Question

我有一个包含三列的文件，比方说，x y z。我需要绘制 x Vs y 但我需要根据其密度(存储在 z 列中)更改该 (x,y) 值的颜色。我知道我需要使用颜色映射并且必须将颜色值映射到 z 数组。我可以通过散点图来做到这一点，如这篇文章中所示:How can I make a scatter plot colored by density in matplotlib?

但我不需要散点图，我需要连接点，即我需要一个线图。可以在线图中完成吗？

Answer 1

无法直接连接散点图中的点。但是在散点后面画一条线也可以达到同样的效果。

import numpy as np
import matplotlib.pyplot as plt

x = np.linspace(-3,6)
y = np.sin(x)
z = 0.5+np.random.rand(len(x))

fig, ax = plt.subplots()
ax.plot(x, y, color="k", marker=None, zorder=0)
sc = ax.scatter(x, y, c=z, s=100, edgecolor='',zorder=3)
plt.colorbar(sc, label="Density")

plt.show()