从条件开始计数行

Question

我在Pandas中有一个包含布尔值的列，并希望计算自上一个True值以来的thr行，如下所示：

a           b
False       0
True        0
False       1
False       2 
False       3  
True        0
False       1
True        0

我可以通过循环来实现，但似乎必须有更好的方法

Answer 1

a = ~df['a']
b = a.cumsum()
c = b-b.where(~a).ffill().fillna(1).astype(int)
print (c)
0    0
1    0
2    1
3    2
4    3
5    0
6    1
7    0
Name: a, dtype: int32

Answer 2

我发现上面的代码不起作用，但是可以做到这一点：

#Create DataFrame
import pandas as pd
d = {'condition': [False,False,False,True,False,False,True,False,True,False,False,False,False,False,False,True]}
df = pd.DataFrame(data=d)

#Calculate Rows Since Condition
df['counter'] = df.index.where(df.condition)
df['counter'].fillna(method="ffill",inplace=True)
df['Rows_since_condition'] = df.index-df['counter']
df.drop(['counter'], axis=1,inplace=True)

#Print Results
df

输出：