spearman相关性分析（R语言findcorrelation()函数的python平替）

直接放代码

import seaborn as sns
import pandas as pd
# Spearman 相关性分析
corr = data.corr(method='spearman')
print(corr) 

# 绘制相关系数热力图
sns.heatmap(corr, annot=True, cmap="YlGnBu") # 特征很多的话这个图会很丑不建议画

# 剔除相关性系数高于threshold的corr_drop
threshold = 0.9 # 自己设置阈值
upper = corr.where(np.triu(np.ones(corr.shape), k=1).astype(np.bool))
corr_drop = [column for column in upper.columns if any(upper[column].abs() > threshold)] # 返回被剔除变量的列表，后续操作省略

虽然是粗糙平替，但必不可能安装R！