将注释和图例保留在相同的seaborn图中

Question

我正在使用seaborn来生成一些数字。

import seaborn as sns
g=sns.JointGrid(...)

最后，我需要添加一个图例并注释该图。我这样做：

...
g.ax_joint.legend(...)
...
g.annotate(scipy.stats.spearmanr,fontsize=14)

但annotate()传说之后不再存在。如何在同一图中将两者保持在同一个ax_joint上？

Answer 1

g.annotate将注释信息添加为图例。在已有图例的绘图中添加新图例将替换旧图例。解决方案是将旧图例读入图中。

oldlegend = plt.legend(<something>)
newlegend = plt.legend(<something else>)
plt.gca().add_artist(legend)

对于这种情况，它看起来像

import seaborn as sns
import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
import scipy.stats

mydataset=pd.DataFrame(data=np.random.rand(50,2),columns=['a','b'])
g = sns.JointGrid(x=mydataset['a'], y=mydataset['b'])
g=g.plot_marginals(sns.distplot,color='black',
                   kde=True,hist=False,rug=True,bins=20)
g=g.plot_joint(plt.scatter,label='X')        


legend_properties = {'weight':'bold','size':8}
legendMain=g.ax_joint.legend(prop=legend_properties,loc='upper right')


legendSide=g.ax_marg_x.legend(labels=["x"], 
                              prop=legend_properties,loc='upper right')
g.annotate(scipy.stats.spearmanr,fontsize=14, loc=4)
g.ax_joint.add_artist(legendMain)
plt.show()