python - 可自定义的字符串热图

Question

我给你写信是因为我没有找到满意的答案。具体来说，我有一个熊猫数据框，其中包含每个变量的字符串字符。制作如下：

Own AUSTRIA.  Own BELGIUM. 

"-1.3"          "-0.34"
"-0.43"         "-1.89**"
"-1.2**"        "-4.5"
"-1.9"          "-2.3"
"-2**"          "-6.1**"
"-.7"           "-0.3"
"-0.06"         "-7.2**"
...             ...
"-1.1**"        "-10.34"

我的目标是生成一个热图，其中具有字符的术语"**"用红色着色，而其他的用蓝色（其他颜色很好）。我知道热图是基于值的输入，但是如果我尝试使用"**"其他值重新调整值，因为我设置的值太高（低），所以热图了解哪些值需要着色。

谢谢，

费德里科

标签： pythonpandasheatmap

Answer 1

在“热图数据框”中，对红色使用大值，对蓝色使用低值。

heatmap = df.copy()
heatmap = heatmap.apply(lambda x: 255 if '**' in x else 0)

你当然可以使用任何你想要的范围，比如 [0, 1], [0, 255], [-1, 1]。