1)如果您的数据是您拥有的宽格式,类别作为索引,列作为值,如下所示:
+--------------------------+
| colA colB colC |
+--------------------------+
| group1 10 5.000 1.200 |
| group2 12 3.000 4.500 |
| group3 14 1.200 2.300 |
+--------------------------+
然后你只需df.heatmap.hvplot()
使用 hvplot >= 0.5:
import pandas as pd
import holoviews as hv
import hvplot.pandas
hv.extension('bokeh')
df = pd.DataFrame({
'colA': [10, 12, 14],
'colB': [5, 3.0, 1.2],
'colC': [1.2, 4.5, 2.3]},
index=['group1', 'group2', 'group3'],
)
df.hvplot.heatmap()
如果您想在热图中添加数据标签,您可以执行以下操作:
heatmap = df.hvplot.heatmap()
heatmap * hv.Labels(heatmap)
2)但是,当您的数据是这样的,其中组只是另一列而不是索引时:
+------------------------------+
| group colA colB colC |
+------------------------------+
| 1 group1 10 5.000 1.200 |
| 2 group2 12 3.000 4.500 |
| 3 group3 14 1.200 2.300 |
+------------------------------+
然后您可以将您的组设置为您的索引df.set_index('group')
(并应用解决方案 1),或者将您的数据融合为长格式:
df_melt = df.melt(id_vars='group')
融化后的数据如下所示:
+---+--------+----------+--------+
| | group | variable | value |
+---+--------+----------+--------+
| 0 | group1 | colA | 10.000 |
| 1 | group2 | colA | 12.000 |
| 2 | group3 | colA | 14.000 |
| 3 | group1 | colB | 5.000 |
| 4 | group2 | colB | 3.000 |
+---+--------+----------+--------+
此融合数据的格式可以使用 x 和 y 以及 C 关键字:
df_melt.hvplot.heatmap(x='group', y='variable', C='value')
或者,您可以使用融化的(长)数据在 HoloViews 中创建热图:
hv.HeatMap(df_melt, kdims=['group', 'variable'], vdims=['value'])
融合数据的优势在于您现在还可以轻松地将数据标签添加到热图中:
heatmap = df_melt.hvplot.heatmap(x='group', y='variable', C='value')
labels = hv.Labels(data=df_melt, kdims=['group', 'variable'], vdims=['value'])
heatmap * labels
将数据标签/值添加到热图中的另一种(甚至)更简单的方法是这样的:
heatmap = df_melt.hvplot.heatmap(x='group', y='variable', C='value')
heatmap * hv.Labels(heatmap)
结果图:
有关 hvplot 中热图的更多信息:
https ://hvplot.holoviz.org/reference/pandas/heatmap.html
有关 holoviews 中热图的更多信息:
https ://holoviews.org/reference/elements/bokeh/HeatMap.html
有关 holoviews 中(数据)标签的更多信息:
https ://holoviews.org/reference/elements/bokeh/Labels.html