Добавьте дополнительные оси к карте тепла морского побережья

Question

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

Я пытался использовать fig.add_axes, но я немного растерялся относительно того, какие параметры и какие значения использовать. Приветствия

parameter = 'Total Rent'
recordtpye = 'Tenant'

dataset = df[(df['Account Record Type'] == recordtpye)]
grouped = dataset.groupby(['TenanatCohort','BookingPeriod'])

cohorts = grouped.agg({'Request ID': pd.Series.nunique,'Total Rent':np.sum})
cohorts.rename(columns={'Request ID': 'Number of Bookings'}, inplace=True)
cohorts.reset_index(inplace=True)
cohorts.set_index(['TenanatCohort', 'BookingPeriod'], inplace=True)

cohort_group_size = dataset.groupby('TenanatCohort').agg({'Tenant: Account Name': pd.Series.nunique})

user_retention_numbers = cohorts[parameter].unstack(0)

bookings_per_cohort = \
cohorts.reset_index()\
.groupby('TenanatCohort')\
.agg({parameter: np.sum})[parameter]\
.divide(cohort_group_size['Tenant: Account Name'])

inp = user_retention_numbers

sns.set(style='white')

fig = plt.figure(figsize=(16,7))

graph = sns.heatmap(inp.T,
                    mask=inp.T.isnull(),
                    cmap="Blues",
                    annot=True,
                    fmt=".0f",
                    annot_kws={"size":10});
graph.set_xlabel("Booking Period")
graph.set_ylabel("Cohort Group")

plt.yticks(rotation=0)

Answer 1

seaborn's heatmap - это функция уровня оси, что означает, что вы можете заранее создать любые оси, которые хотите, и передать ссылку на объект Axes, который seaborn будет использовать при вызове heatmap.

uniform_data = np.random.rand(10, 12)
fig, (ax1, ax2) = plt.subplots(1,2, figsize=(6,4), gridspec_kw={'width_ratios':(5,1)})
sns.heatmap(uniform_data, ax=ax1)
ax2.plot([1,2],[3,4], 'ro-')