floats ------------------------ (1.2, 3.4, 5.6) [重复]

Question

这个问题在这里已经有了答案:

如何在matplotlib中把数字转换为色标？ (1个回答)

封闭 21天之前.

我有一个数据框，其中一列是numpy数组，另一列是浮动数组，像这样。

arrays          | floats
------------------------
(1.2, 3.4, 5.6) |  4.5
(1.7, 4.4, 8.1) |  5.5

我想把每一个数组都绘制在一个图中 但我需要线条的颜色取决于与之相关的浮点。

到目前为止，我已经做到了这一点。

# Plot
import matplotlib.pyplot as plt
import matplotlib.colors as mcolors
import matplotlib.cm as cm

colormap = cm.jet
normalize = mcolors.Normalize(vmin=np.min(floats.values), vmax=np.max(floats.values))
s_map = cm.ScalarMappable(norm=normalize, cmap=colormap)

x_axis = np.linspace(0, 10, len(arrays.values[0]))

for arr, flt in in zip(arrays.values, floats.values):
   plt.plot(x_axis, arr)
plt.show()

我知道可以传递一个颜色..: plt.plot(x_axis, arr, color='b')但我如何使用 s_map 根据第二列的值来做这个变量？

我在这里读到了类似的问题，但没有一个能解决我的问题。