[閒聊] 如何讓值在一段範圍內一直+1

作者: koow ( )   2021-01-24 20:15:56
如題 請問一下各位前輩
我想要讓dataframe columns資料中
在某一段時間一直持續+1
ex:
trigger hold
0 0 0
1 0 0
2 1 0
3 0 1
4 0 2
5 0 3
6 0 4
7 -1 5
有一個trigger資料1的時候
hold下一列開始持續+1直到 trigger有-1出現就停止
原本想說很單純用
df.loc[df.trigger.shift(1)==1,['hold']]=1
df.loc[df.hold.shift(1)!=0,['hold']]=df.hold.shift(1)+1
但第一行可以出現我要的
但第二行不行
可否請問一下前輩問題出在哪
或是有甚麼建議的寫法呢
感謝
作者: chickengod (雞大俠)   2021-01-25 11:13:00
df.hold = df.trigger.cumsum()df.hold = df.hold.cumsum().shift(1).fillna(0)
作者: ee24584552   2021-02-01 23:48:00
df['hold'] = df['trigger'].cumsum().cumsum()好像不太一樣XD 請忽略

Links booklink

Contact Us: admin [ a t ] ucptt.com