Pandas - 自上次大於0的行數

Question

假設我有一個像這樣的熊貓系列：

import pandas as pd

pd.Series([1, 0, 0, 1, 0, 0, 0], name='series')

自上次> 0后，我如何添加一個包含行數的列，如下所示：

pd.DataFrame({
    'series': [1, 0, 0, 1, 0, 0, 0],
    'row_num': [0, 1, 2, 0, 1, 2, 3]
})

Answer 1

試試這個：

s.groupby(s.cumsum()).cumcount()

輸出：

0    0
1    1
2    2
3    0
4    1
5    2
6    3
dtype: int64

Answer 2

NumPy的

• 找到系列/數組大於0的位置
• 計算從一個地方到下一個地方的差異
• 從序列中減去這些值

---

i = np.flatnonzero(s)
n = len(s)
delta = np.diff(np.append(i, n))
r = np.arange(n)
r - r[i].repeat(delta)

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