python - Seaborn clustermap 作为子图

Question

seaborn clustermap似乎是一个图形级别的情节，它在内部创建自己的图形和轴。

我想将集群用作子图，以便能够在同一图形上添加额外的图(例如，热图顶部的箱线图)。

我怎样才能做到这一点？

谢谢

聚类图示例:

import numpy as np
import pandas as pd
import seaborn as sns

nb_features = 20

tmp = [v for v in zip(np.repeat(['A', 'B'], 5), 
       np.hstack([range(5), range(5)]))]
index = pd.MultiIndex.from_tuples(tmp)

df = pd.DataFrame(np.vstack([np.random.rand(5, nb_features) + 1, 
                             np.random.rand(5, nb_features) - 1]), index=index).T

# Returns a seaborn.matrix.ClusterGrid
cax = sns.clustermap(df)

Answer 1

您可以在图中添加一些空间，例如通过 subplots_adjust()。然后您可以通过 fig.add_axes([x, y, width, height]) 添加一个新轴(在 figure coordinates 中表示)。

这是一个例子:

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

sns.set_theme(color_codes=True)
iris = sns.load_dataset("iris")
species = iris.pop("species")
g = sns.clustermap(iris)
g.fig.set_size_inches(20, 10)
g.fig.subplots_adjust(right=0.5)
ax = g.fig.add_axes([0.55, 0.05, 0.42, 0.92])
ax.plot(np.random.randn(1000).cumsum())
plt.plot()