基于bool系列高效分配到熊猫系列

时间:2017-03-20 17:44:16

标签: python python-3.x pandas boolean series

给定Series个bool,我想创建一个从0开始的新Series,并在布尔Series中的每个相应条目递增True 。这是一个例子:

s = pd.Series([False, True , False, False, False, True, False, False, True, False])

看起来像:

0    False
1     True
2    False
3    False
4    False
5     True
6    False
7    False
8     True
9     False

我正在寻找一种有效的,如果可能的优雅操作,而不需要循环,以获得以下Series

0    0
1    1
2    1
3    1
4    1
5    2
6    2
7    2
8    3
9    3

1 个答案:

答案 0 :(得分:1)

使用pd.Series.cumsum

s.cumsum()

0    0
1    1
2    1
3    1
4    1
5    2
6    2
7    2
8    3
9    3
dtype: int64