python - 如何获取指定值的最后一行的排名？

Question

我正在研究这样的数据框:

group | rank |  boolean
-----------------------
1     |1    |  True
1     |2    |  False
1     |3    |  False
2     |1    |  False
2     |2    |  True
2     |3    |  False

我希望对于每一行，获得同一组中 bool 列中具有真值的最后一个排名:

group | rank |  boolean | output
--------------------------------
1     |1    |  True     |1
1     |2    |  False    |1
1     |3    |  False    |1
2     |1    |  False    |null
2     |2    |  True     |2
2     |3    |  False    |2

Answer 1

我确信有一种更简洁的方法可以做到这一点，但是这个函数应该可以满足您的要求

def new_column(df):
    to_iterate_over=df.group.unique()

    g = df.groupby('group')
    new_column_to_add=[]
    for x in to_iterate_over:        
        value='Null'
        for index, row in g.get_group(x).iterrows():
            if row['boolean']==True:
                value=row['rank']
            new_column_to_add.append(value)

    df['output']=new_column_to_add
    return df

df_new=new_column(df)

当我运行它时，我得到了正确的输出。不过，我会研究更简洁的实现