为seaborn热图添加一个额外的轴

Question

我有一个保留矩阵作为seaborn热图。我想知道，我在哪里以及如何添加额外的轴。我希望在热图旁边有一个额外的垂直条形图，以显示给定同类群组中的新用户数。任何帮助或指示将非常感激。

我尝试使用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的heatmap是一个轴级功能，这意味着你可以事先创建你想要的任何轴，并在调用heatmap时传递一个你希望seaborn使用的Axes对象的引用。

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-')