根据下一行的值更新当前行

Question

我有以下数据框：

df = pd.DataFrame({'ID': ['1','1','1','1','2','2','2'], \
                   'episode': ['111', '112','113', '114','115','116','117'], \
                   'risk': ['L', 'L','H', 'L','H','L','L'], \
                   'date': ['21-01-2015', '27-02-2015','19-03-2015', '09-05-2015','3-01-2015','7-02-2015','11-05-2015']})

每个ID可以有几行（不同的剧集编号），每一行都有一个风险标识为L或H。我想检查是否每个ID都存在风险H。一旦有风险H 出现在 ID 的任何剧集中，我想将剩余行中的风险更改为 H，以便同一 ID 的任何先前剧集都应具有风险 H。数据已按 ID、剧集和日期排序。最终输出将是：

  ID       date episode risk
0  1 2015-01-21     111    L
1  1 2015-02-27     112    L
2  1 2015-03-19     113    H
3  1 2015-05-09     114    H
4  2 2015-01-03     115    H
5  2 2015-02-07     116    H
6  2 2015-05-11     117    H

我怎样才能做到这一点？

Answer 1

检查 groupby + cumprod

df['risk'].ne('H').groupby(df['ID']).cumprod().map({True:'L',False:'H'})
Out[265]: 
0    L
1    L
2    H
3    H
4    H
5    H
6    H
Name: risk, dtype: object
#df['risk']=    df['risk'].ne('H').groupby(df['ID']).cumprod().map({True:'L',False:'H'})

根据下一行的值更新当前行

update current row based on values of next row

pandas

python-3.5

pandas-groupby