我在Pandas中有一个包含布尔值的列,想要计算自最后一个True值以来的行数,类似于这样:
a b
False 0
True 0
False 1
False 2
False 3
True 0
False 1
True 0
我可以使用循环来完成这个任务,但是似乎还有更好的方法
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
#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
输出: