python - 获取行和列 pandas bool 数据框

Question

我有一个 bool Pandas数据框(所有值都是True或False)，我需要获取每一行的列其中值为 True。

知道我该怎么做吗？

Answer 1

使用[np.where(row)[0]作为索引，df.iterrows()中的行]它将生成一个 numpy 数组列表，每行中都有 True 值的索引。添加 .tolist() 作为列表列表。

无法堆叠结果数组，因为它们可能具有不同的长度。

示例:

df = pd.DataFrame([[True, False],[False,True],[True,True], [False,False]])
[np.where(row)[0].tolist() for  _, row in df.iterrows()]

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