自条件以来计算行数

Question

我在Pandas中有一个包含布尔值的列，想要计算自最后一个True值以来的行数，类似于这样：

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

我可以使用循环来完成这个任务，但是似乎还有更好的方法

- gdnaes

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

- jezrael

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

输出：

- cwse