相关矩阵：提取具有高 R 值的变量答案

【问题标题】：Correlation Matrix: Extract Variables with High R Values相关矩阵：提取具有高 R 值的变量
【发布时间】：2016-06-09 00:24:03
【问题描述】：

如何获得仅列出绝对值相关性大于 0.7 的变量的输出？

我想要类似这样的输出：

four: one, three
one: three

感谢您的宝贵时间！

代码

import pandas as pd

x={'one':[1,2,3,4],'two':[3,5,7,5],'three':[2,3,4,9],'four':[4,3,1,0],}
y=pd.DataFrame(x)
print(y.corr())

输出

           four       one     three       two
four   1.000000 -0.989949 -0.880830 -0.670820
one   -0.989949  1.000000  0.913500  0.632456
three -0.880830  0.913500  1.000000  0.262613
two   -0.670820  0.632456  0.262613  1.000000

【问题讨论】：

标签： python matrix correlation

【解决方案1】：

如果您只想将其打印出来，这将起作用：

col_names = y.corr().columns.values

for col, row in (y.corr().abs() > 0.7).iteritems():
    print(col, col_names[row.values])

请注意，这可行，但可能会很慢，因为iteritems 方法会将每一行转换为一个系列。

【讨论】：

谢谢 juanpa.arrivillaga。
我尝试编辑您的代码以排除 r=1，但我收到一条错误消息。您将如何编辑此代码？对于 col, row in (y.corr().abs() > 0.7).iteritems()：这对 col, row in (y.corr().abs() > 0.7 和 y 不起作用。 corr().abs()
啊。因此，如果您想对 pandas 对象进行 element-wise 逻辑操作，您需要使用 & 对应 and 和 | 对应 or。所以这应该有效：(y.corr().abs() > 0.7) & (y.corr().abs() < 1)。看到这个答案：stackoverflow.com/questions/21415661/…
再次感谢！你解决了我的问题，我也从你的所作所为中学到了东西。我一定会付清的。

【解决方案2】：

这对我有用：

corr = y.corr().unstack().reset_index() #group together pairwise
corr.columns = ['var1','var2','corr'] #rename columns to something readable
print( corr[ corr['corr'].abs() > 0.7 ] ) #keep correlation results above 0.7

您可以通过将最后一行更改为

来进一步排除具有相同名称 (corr = 1) 的变量

print( corr[ (corr['corr'].abs() > 0.7) & (corr['var1'] != corr['var2']) ] )

【讨论】：