python-3.x - 打印相关性大于 80% 的列

Question

我有一个大小为 235607 records 的 Pandas 数据框, 和 94 attributes .我很新 python我能够在所有属性之间创建一个相关矩阵，但要单独查看很多。我尝试编写一个 for 循环来打印相关性大于 80% 的列的列表，但我不断收到错误“'DataFrame' 对象没有属性 'c1'”

这是我用来创建属性之间相关性以及循环示例的代码。预先感谢您的帮助 :-

corr = data.corr() # data is the pandas dataframe
c1 = corr.abs().unstack()
c1.sort_values(ascending = False)
drop = [cols for cols in upper.c1 if any (upper[c1] > 0.80)]
drop

Answer 1

您可以简单地使用 numpy.where像这样:

corr.loc[np.where(corr>0.8, 1, 0)==1].columns

输出将是列名的数组，列名的值大于 0.8 .

编辑:我希望这会奏效。我编辑了上面的代码。