python - 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]
})

标签： pythonpandasindexing

Answer 1

尝试这个：

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

输出：

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