python - 在 Pandas 中获取 bool 数据帧的 True 元素的(索引，列)对

Question

假设我有一个 Pandas DataFrame，我想获得一个形式为 [(index1, column1), (index2, column2) ...] 的元组列表，描述 DataFrame 中所有元素的位置，其中某些条件是真的。例如:

x = pd.DataFrame(np.random.normal(0, 1, (4,4)), index=['a', 'b', 'c', 'd'],
                 columns=['e', 'f', 'g', 'h'])
x


     e           f           g           h
a   -1.342571   -0.274879   -0.903354   -1.458702
b   -1.521502   -1.135800   -1.147913   1.829485
c   -1.199857   0.458135    -1.993701   -0.878301
d   0.485599    0.286608    -0.436289   -0.390755

y = x > 0

有什么办法可以得到:

x.loc[y]

返回:

[(b, h), (c,f), (d, e), (d,f)]

或者一些等价物？显然，我可以这样做:

postup = []
for i in x.index:
    for j in x.columns:
        if x.loc[i, j] > 0:
            postup.append((i, j))

但我想可能会有更好的东西/已经实现了。在 matlab 中，函数 find 结合 sub2ind 完成这项工作。

Answer 1

x[x > 0].stack().index.tolist()