根据该组中列的第一行值更改分组 dataframe 中的值

Question

我正在使用下面的代码来安排一些数据。 它按'ID'和'L'分组，如果它是True ，我已经使用条件在MT列中打印 True 。 我想要的是，当该组的第一行为真时，该组中的每一行都应打印为真，而不管条件如何（例如在下面的 output 中，MT 列中的第五行也应该显示真，因为第一行评估为真）。

import pandas as pd
df=pd.DataFrame({'TS' : [20,25,30,35,45,50,55,58,65,75,60,62,72,77],"ID" : [22,22,23,23,23,23,23,23,23,23,24,24,24,24],"L" : [1,2,1,2,2,2,1,1,2,2,1,1,2,2]})
df2=(df.assign(TS=df['TS'].abs())
    .groupby(['ID','L'])['TS'].agg([('Min' , 'min'), ('Max', 'max')])
    .add_prefix('TS'))
df3=(df2.assign(WT=df2['TSMax']-df2['TSMin'])
    .assign(MT=df2['TSMax'].gt(df2.groupby('ID')['TSMin'].shift(-1))))
df3

Output：

Answer 1

您可以使用groupby ，在 MT 列上建立索引，然后使用transform检查值是否为 True 进行转换。

df3['MT'] = df3.groupby('ID')['MT'].transform(lambda x: x.values[0] if x.values[0] else x)

Output：