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

您可以在图中添加一些space，例如通过 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()

Seaborn clustermap 作为子图

Seaborn clustermap as a subplot

python

pandas

seaborn