Seaborn clustermap作为子图

时间:2016-08-06 03:32:34

标签: python pandas seaborn

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)

enter image description here

1 个答案:

答案 0 :(得分: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()

example clustermap with new subplot