在python中的数据框中计算与引用可变值的条件相匹配的行

Posted 2021-04-01

我正在尝试在数据框BoughtProducts中创建一列CustomerPurchases，其中将包含该客户先前的购买次数。数据框包含订单日期（OrderDateTime）和客户编号（CustomerNumber）等，我想为每一行计数，数据框中有多少行具有相同客户编号和较早订购日期的行。我试图那样做：

for index, row in BoughtProducts.iterrows():

  row['CustomersPurchases']=BoughtProducts[(BoughtProducts['OrderDateTime']<=row['OrderDateTime'])&(BoughtProducts['CustomerNumber']==row['CustomerNumber'])].count()

但是运行循环需要花很多时间，我相信有一种更简便，更快速的方法。

答案

可以使用sort_values，groupby和cumcount功能来完成。

BoughtProducts['CustomersPurchases'] = (
    BoughtProducts
    .sort_values(by['OrderDateTime']))
    .groupby('CustomerNumber')
    .cumcount()+1
)