pandas实现相关性最强的两个变量

原数据：

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')

'''

得到相关性最强的两列的标签。