原数据:
import pandas as pd
df=pd.DataFrame({'A':[1,2,4,5,6],'B':[2,4,6,9,10],'C':[2,1,7,2,1]})
df
'''
A B C
0 1 2 2
1 2 4 1
2 4 6 7
3 5 9 2
4 6 10 1
'''
实现相关性可以用到df.corr()得到相关矩阵,相关性系数从-1到1表示相关程度。
(
df.corr()
.stack()
.sort_values(ascending=False)
.loc[lambda x:x<1]
.idxmax()
)
'''
('A', 'B')
'''
得到相关性最强的两列的标签。