python - 根据条件的最后一次出现划分数据帧

Question

假设我有按 id 排序的数据:

 id | count 
 1    1
 2    2
 3    0
 4    4
 5    3
 6    2
 7    0
 8    10
 9    1
 10   2

我想始终获取任何最后 零 之后的最后更改。根据以上数据，我想得到:

 id | count 
 8    10
 9    1
 10   2

有人知道怎么做吗？

Answer 1

Pandas

df.loc[df['count'].ne(0).iloc[::-1].cumprod().astype(bool)]

   id  count
7   8     10
8   9      1
9  10      2

numpy

df[(df['count'].values[::-1] != 0).cumprod()[::-1].astype(bool)]

   id  count
7   8     10
8   9      1
9  10      2

---

有其他条件

df[(df['count'].values[::-1] < 3).cumprod()[::-1].astype(bool)]
# df.loc[df['count'].lt(3).iloc[::-1].cumprod().astype(bool)]

   id  count
8   9      1
9  10      2

---

调试
您应该能够复制并粘贴它并重现我的结果。如果你不能，那么还有其他问题。尝试重置内核。

import pandas as pd

df = pd.DataFrame({
        'count': [1, 2, 0, 4, 3, 2, 0, 10, 1, 2],
        'id': [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
    })

df[(df['count'].values[::-1] < 3).cumprod()[::-1].astype(bool)]

---

应该产生

   count  id
8      1   9
9      2  10