python - Pandas 根据条件返回索引和列名

Question

假设我们有一个 DataFrame:

df = pd.DataFrame({
    'D': {'A': 0.810032, 'B': 0.752299, 'C': 0.224038},
    'E': {'A': 0.17714, 'B': 0.680405, 'C': 0.769732},
    'F': {'A': 0.942959, 'B': 0.436582, 'C': 0.269791}
})
print(df)
          D         E         F
A  0.810032  0.177140  0.942959
B  0.752299  0.680405  0.436582
C  0.224038  0.769732  0.269791

是否可以根据条件将索引和列名返回到元组列表中，例如

print(df < 0.25)
   D      E      F
A  False   True  False
B  False  False  False
C  True  False  False

我想以这样的元组列表结束:

[('A','E'),('C','D')]

Answer 1

我们也可以使用DataFrame.where然后 stack它利用了 stack 的事实默认情况下删除 NaN 值然后使用 Index.tolist获取元组列表:

results = df.where(df < .25).stack().index.tolist()

结果:

[('A', 'E'), ('C', 'D')]