有没有一种方法可以根据值从熊猫DataFrame中提取索引

Question

我有以下DataFrame：

index col0 col1 col2
0     0    1    0
1     1    0    1
2     0    1    1

我想提取以下索引（那些包含一个（或任何值）的索引）：

[(0, 1), (1, 0), (1, 2), (2, 1), (2,2))]

大熊猫中有一种方法可以做到吗？

Answer 1

使用np.where + zip

---

[*zip(*np.where(df))]

[(0, 1), (1, 0), (1, 2), (2, 1), (2, 2)]

Answer 2

这是一种方法

df.columns=np.arange(df.shape[1])
df.stack().loc[lambda x : x==1].index.tolist()
[(0, 1), (1, 0), (1, 2), (2, 1), (2, 2)]

Answer 3

您可以使用numpy.nonzero：

result = list(zip(*np.nonzero(df.values)))
print(result)

输出

[(0, 1), (1, 0), (1, 2), (2, 1), (2, 2)]