我有一个熊猫据帧school_df
,看起来像这样:通过迭代大熊猫GROUPBY组
school_id date_posted date_completed
0 A 2014-01-01 2014-01-01
1 A 2014-01-01 2014-01-08
2 A 2014-04-29 2014-05-01
3 B 2014-01-01 2014-01-01
4 B 2014-01-20 2014-02-23
每一行代表由学校一个项目。我想添加两列:对于每个唯一的school_id
,计算在该日期之前发布的项目数量以及在该日期之前完成了多少项目的计数。
下面的代码有效,但我有大约300,000个独特的学校,所以需要很长时间才能运行。有没有更快的方式来获得我正在寻找的东西?谢谢您的帮助!
import pandas as pd
groups = school_df.groupby("school_id")
blank_df = pd.DataFrame()
for g, df in groups:
df['school_previous_projects'] = df.date_posted.map(lambda x: len(df[df.date_posted < x]))
df['school_previous_completed'] = df.date_posted.map(lambda x: len(df[df.date_completed < x]))
blank_df = pd.concat([blank_df, df])
@BobHaffner有一个很好的答案。在盒子外面思考,你可以分组学校,并在日期栏中一次设置索引。然后你可以使用滚动计数,因为它将按日期排序。这比使用apply方法和检查每行的len要快得多。查看cumcount http://pandas.pydata.org/pandas-docs/stable/generated/pandas.core.groupby.GroupBy.cumcount.html –
我同意@BrianPendleton我的方法可能会比您的方法更快,但可能会有一个更好的方法。 –