python-3.x - 获取与数字的一维热图对应的颜色

Question

对于某些给定的数字，我想获得相应的颜色，其强度随着数字的增加或减少而增加或减少。

我有一个这样的数据框：

data={'A':1,'B':0.5,'C':0.3,'D':1.2}
df=pd.DataFrame(data,index=[0])
df=pd.melt(df,var_name="Tokens")
df.head()


  Tokens    value
0   A       1.0
1   B       0.5
2   C       0.3
3   D       1.2

我想获得强度随着 df['value'] 中值的增加或减少而变化的颜色。例如，序列 ABCD 的颜色首先是 A 的高强度，然后是 B 和 C 的低强度，然后是 D 的高强度。

我曾尝试使用一维热图，例如：

vec1 = np.asarray([1,2,3,4,5])
fig, ax = plt.subplots()
sn.heatmap([vec1])

但我无法进一步取得进展，因为我不知道以后如何获得与数字相对应的颜色。我想根据值使用与文本对应的颜色突出显示文本。任何帮助将不胜感激。谢谢！

标签： python-3.xheatmap