有沒有一種方法可以根據值從 pandas 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))]

pandas 中是否有可以做到這一點的方法？

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)

Output

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