python - 获取 Pandas 中 boolean 值的索引 - python

Question

a=pd.DataFrame({'a1':[1,2,3,4],'a2':[5,6,7,8]})
c=pd.DataFrame({'c1':[True,False,True,True],'c2':[True,False,False,True]})

如何获取 a1 和 a2 列中元素的索引，这些元素在 c1 列中为 True > 和 c2 分别是？

a1 的索引应该是[0,2,3] 而a2 的索引应该是[0,3] 。结果可能是一个索引列表，如 [[0,2,3],[0,3]]。

Answer 1

我想你需要where :

c.columns = a.columns
df1 = a.where(c)
idx = [df1[x].dropna().index.tolist() for x in df1]
print (idx)
[[0, 2, 3], [0, 3]]

另一种解决方案:

c.columns = a.columns
idx = list(a.where(c).stack().reset_index()
            .groupby('level_1')['level_0'].apply(list))
print (idx)
[[0, 2, 3], [0, 3]]