我的 df：import pandas as pdimport numpy as npdf = pd.DataFrame({'id':[1,1,1,2,2], 'time':['2020-01-01 12:00:15','2020-01-01 12:00:30','2020-01-01 12:00:45','2020-01-03 08:00:00','2020-01-03 08:00:15'], 'time1':['2020-01-01 12:00:00','2020-01-01 12:00:00','2020-01-01 12:00:00','2020-01-01 12:00:00','2020-01-01 12:00:00'], 'numb':[1,5,8,0,4]})df['time'] = pd.to_datetime(df['time'])df['time1'] = pd.to_datetime(df['time1'])df['numb_diff'] = df['numb'] - df['numb'].shift()輸出： id time time1 numb numb_diff0 1 2020-01-01 12:00:15 2020-01-01 12:00:00 1 NaN1 1 2020-01-01 12:00:30 2020-01-01 12:00:00 5 4.02 1 2020-01-01 12:00:45 2020-01-01 12:00:00 8 3.03 2 2020-01-03 08:00:00 2020-01-01 12:00:00 0 -8.04 2 2020-01-03 08:00:15 2020-01-01 12:00:00 4 4.0現(xiàn)在我想設(shè)置為組（ ）time1的最低值，只要該位置的第一個(gè)條目<0。timeididnumb_diff預(yù)期輸出： id time time1 numb numb_diff0 1 2020-01-01 12:00:15 2020-01-01 12:00:00 1 NaN1 1 2020-01-01 12:00:30 2020-01-01 12:00:00 5 4.02 1 2020-01-01 12:00:45 2020-01-01 12:00:00 8 3.03 2 2020-01-03 08:00:00 2020-01-03 08:00:00 0 -8.0 #Changing time1 to the min of time the group(id = 2)4 2 2020-01-03 08:00:15 2020-01-03 08:00:00 4 4.0
1 回答
慕姐4208626
TA貢獻(xiàn)1852條經(jīng)驗(yàn) 獲得超7個(gè)贊
讓我們創(chuàng)建一個(gè)表示條件的布爾掩碼，其中num_diff小于零，然后對(duì)該掩碼進(jìn)行分組id并使用 進(jìn)行轉(zhuǎn)換first，最后使用此掩碼的布爾索引來(lái)替換 中的值time1：
m = df['numb_diff'].lt(0).groupby(df['id']).transform('first')
df.loc[m, 'time1'] = df.groupby('id')['time'].transform('min')
id time time1 numb numb_diff
0 1 2020-01-01 12:00:15 2020-01-01 12:00:00 1 NaN
1 1 2020-01-01 12:00:30 2020-01-01 12:00:00 5 4.0
2 1 2020-01-01 12:00:45 2020-01-01 12:00:00 8 3.0
3 2 2020-01-03 08:00:00 2020-01-03 08:00:00 0 -8.0
4 2 2020-01-03 08:00:15 2020-01-03 08:00:00 4 4.0
添加回答
舉報(bào)
0/150
提交
取消