Pandas

Question

の前のセルの値にカウントを追加する別の列 'A'のブール値に応じて、ある列 'B'に値を追加しようとしています。したがって、 'A'が真であれば、 'A'が偽である限り、カウントを開始します（つまり、それぞれ新しい行を1つ追加します）。 'A'がTrueの場合はリセットし、再度カウントを開始します。私は 'for'ループでこれを行うことができましたが、これは非常に時間がかかります。効率的な解決策がなくなるのではないかと疑問に思っていますか？

結果は次のようになります。

Date  A  B 
01.2010 False 0 
02.2010 True 1 
03.2010 False 2 
04.2010 False 3 
05.2010 True 1 
06.2010 False 2 

Answer 1

あなたがgroupbyとcumcountでcumsumを使用することができます。

print df 
    Date  A 
0 1.201 False 
1 1.201 True 
2 1.201 False 
3 2.201 True 
4 3.201 False 
5 4.201 False 
6 5.201 True 
7 6.201 False 
roll = df.A.cumsum() 
print roll 
0 0 
1 1 
2 1 
3 2 
4 2 
5 2 
6 3 
7 3 
Name: A, dtype: int32 

df['B'] = df.groupby(roll).cumcount() + 1 
#if in first values are False, output is 0 
df.loc[roll == 0 , 'B'] = 0 
print df 
    Date  A B 
0 1.201 False 0 
1 1.201 True 1 
2 1.201 False 2 
3 2.201 True 1 
4 3.201 False 2 
5 4.201 False 3 
6 5.201 True 1 
7 6.201 False 2 

Answer 2

おかげで、私はこれに似た別のポストから解決策を得た：

rolling_count = 0 
def set_counter(val): 

    if val == False: 

    global rolling_count 
    rolling_count +=1 

    else: 

    val == True 
    rolling_count = 1 

    return rolling_count 

df['B'] = df['A'].map(set_counter)