问题描述
是否有一种方法可以显示seaborn.pairplot()
的成对相关值,如下面的示例(在ggpairs()
中用R
创建)?我可以使用随附的代码进行绘制,但是无法添加相关性。谢谢
import numpy as np
import seaborn as sns
import matplotlib.pyplot as plt
iris = sns.load_dataset('iris')
g = sns.pairplot(iris,kind='scatter',diag_kind='kde')
# remove upper triangle plots
for i,j in zip(*np.triu_indices_from(g.axes,1)):
g.axes[i,j].set_visible(False)
plt.show()
解决方法
如果您使用PairGrid
而不是pairplot
,则可以传递一个自定义函数,该函数将计算相关系数并将其显示在图形上:
from scipy.stats import pearsonr
def reg_coef(x,y,label=None,color=None,**kwargs):
ax = plt.gca()
r,p = pearsonr(x,y)
ax.annotate('r = {:.2f}'.format(r),xy=(0.5,0.5),xycoords='axes fraction',ha='center')
ax.set_axis_off()
iris = sns.load_dataset("iris")
g = sns.PairGrid(iris)
g.map_diag(sns.distplot)
g.map_lower(sns.regplot)
g.map_upper(reg_coef)