Seaborn 2d histplot：显示每个 bin 中的计数

Question

如 seaborn histplot - 在每个条形上方打印 y 值中所述，可以使用

.bar_label(ax.containers[0])

在一维直方图中的每个条形上显示计数。

我正在努力弄清楚如何对二维直方图（使用

sns.histplot(data, x='var1', y='var2')

创建）进行等效操作。

我知道我可以使用

(a,b)

为

.annotate('foo', xy=(a, b))

bin 进行注释，但我不确定如何检索该 bin 的计数（将其传递给

.annotate()

）。

我希望结果与 https://seaborn.pydata.org/examples/spreadsheet_heatmap.html 中显示的结果类似，只是它是一个直方图，而不是热图。

Answer 1

您可以利用

np.histogram2d()

进行计数，然后通过

sns.heatmap()

显示结果：

import matplotlib.pyplot as plt
import seaborn as sns
import numpy as np

np.random.seed(20240529)
x = np.random.randn(1000).cumsum()
y = np.random.randn(1000).cumsum()

hist, xbins, ybins = np.histogram2d(x, y)
xlabels = [f'{x:.2f}' for x in (xbins[:-1] + xbins[1:]) / 2]
ylabels = [f'{y:.2f}' for y in (ybins[:-1] + ybins[1:]) / 2]
sns.heatmap(hist.T, xticklabels=xlabels, yticklabels=ylabels, annot=True, fmt='.0f', cbar=False)