python-3.x - 选择 'student_id'，其中 'pass' 值在每种情况下都不为 0

Question

我有一个数据集。我想选择 student_id 的“pass”值不全为零的所有行

student_id      Subject      pass

234554          Maths        1
346535          Science      0  
862789          History      0
234554          English1     1
346535          History      0
862789          Sciencre     1

对于student_id 346535，没有一个pass值是1，所以忽略

预期输出:

student_id      Subject      pass

234554          Maths        1
862789          History      0
234554          English1     1
862789          Sciencre     1

Answer 1

set_index + 任意

u = df.set_index('student_id')

u.loc[u.groupby(level=0)['pass'].any()]

             Subject  pass
student_id
234554         Maths     1
862789       History     0
234554      English1     1
862789      Sciencre     1

---

any 在这里与 any 一起工作是开箱即用的，因为它是“假的”。如果您正在寻找一个特定值，而不是一个“虚假”值，您可以在groupby 之前进行屏蔽以创建loc 的索引器

u['pass'].ne(val).groupby(level=0).any()