如何在熊猫中添加“群内订单”一栏？

Question

采用以下数据帧：

import pandas as pd
df = pd.DataFrame({'group_name': ['A','A','A','B','B','B'],
                   'timestamp': [4,6,1000,5,8,100],
                   'condition': [True,True,False,True,False,True]})

我想增加两列：

1. 行在其组中的顺序

我知道我可以用自定义应用程序来实现，但我想知道是否有人有什么有趣的想法？（当有许多组时，这也很慢。）这里有一个解决方案：

def range_within_group(input_df):
    df_to_return = input_df.copy()
    df_to_return = df_to_return.sort('timestamp')
    df_to_return['order_within_group'] = range(len(df_to_return))
    df_to_return['rolling_sum_of_condition'] = df_to_return.condition.cumsum()
    return df_to_return

df.groupby('group_name').apply(range_within_group).reset_index(drop=True)

Answer 1

GroupBy.cumcount

将每组中的每个项目从0编号到该组的长度-1。

简单地说：

>>> gr = df.sort('timestamp').groupby('group_name')
>>> df['order_within_group'] = gr.cumcount()
>>> df['rolling_sum_of_condition'] = gr['condition'].cumsum()