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)
答案 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()