python - 获取回答熊猫中过滤器的所有值的索引

Question

我有一个包含浮点数的 DataFrame，我想获取与某个过滤器匹配的所有单元格索引。

所以假设我有这个DataFrame：

	一个	乙	C
一个	1	0.7	0.9
乙	0.7	1	0.3
C	0.9	0.3	1

我的过滤器> = 0.9 我想获得索引（0,0），（1,1），（2,2），（0,2），（2,0）。

或者更具体地说，我有 pearson 相关数据框，我想获得相关性大于 0.9 的所有列

标签： pythonpandas

Answer 1

您可以使用np.argwhere()：

import numpy as np
out=np.argwhere(df.to_numpy()>=0.9).tolist()

输出out：

[[0, 0], [0, 2], [1, 1], [2, 0], [2, 2]]