python - Holoviews 中的热图？

Question

我正在尝试使用 Holoviews 绘制热图。我做了几次尝试但没有成功。我绘制的数据有 2 个值(一个在“x”轴上，一个在“y”轴上)，我希望每种颜色代表数量(就像热图中的直方图)。

Answer 1

在 HoloViews 中，HeatMap 实际上用于分类数据。如果您的数据是数值型的，您确实想计算 2D 直方图并使用 hv.Image 元素，则可以使用 np.histogram2d 来完成此操作，例如:

a, b = np.random.randn(1000, 2).T
df = pd.DataFrame({'a': a*10, 'b': b}, columns=['a', 'b'])
z, a, b = np.histogram2d(df['a'], df['b'])
hv.Image((a, b, z), ['a', 'b'], 'Count')

或者，如果您有大量数据，您可以使用内置的数据着色器支持来执行相同的操作:

from holoviews.operation.datashader import rasterize

a, b = np.random.randn(1000, 2).T
df = pd.DataFrame({'a': a*10, 'b': b}, columns=['a', 'b'])
rasterize(hv.Scatter(df), width=10, height=10, dynamic=False)

根据您的需要调整宽度和高度，或删除dynamic=False以在缩放时动态重新采样。