Как изменить масштаб подзаговоров в Seaborn?

Question

Я создаю фигуру с двумя тепловыми картами

fig, axs = plt.subplots(ncols=2, figsize=(20, 15))
heatmap(data1, cmap=color_palette('Greys_r'), square=True, ax=axs[0])
heatmap(data2, cmap=color_palette('Greys_r'), square=True, ax=axs[1])
fig.savefig('heatmap.png')

Однако полученные тепловые карты слишком малы (или легенды слишком велики)

Я пытался установить figsize в (20, 15), но это не имеет никакого очевидного эффекта. Как я могу это исправить?

Answer 1

Вероятно, уродливый хак, когда вам приходится манипулировать параметром shrink вручную, но его можно использовать для текущей проблемы.

import numpy as np
import seaborn as sns
data1 = np.random.rand(10, 12)
data2 = np.random.rand(10, 12)

fig, axs = plt.subplots(ncols=2, figsize=(20, 15))
sns.heatmap(data1, cmap=sns.color_palette('Greys_r'), square=True, cbar_kws={"shrink": .42}, ax=axs[0])
sns.heatmap(data2, cmap=sns.color_palette('Greys_r'), square=True, cbar_kws={"shrink": .42}, ax=axs[1])

выход