每月按月对大熊猫进行RFM分析

时间:2018-07-30 23:20:51

标签: python-3.x pandas dataframe group-by

我最近一年的交易中对熊猫进行了RFM分析,但是现在我想给每个客户每个月的RFM得分,我的数据框如下所示:

txn_id | customer_id | date     | total
1      | 2           | 2016-1-2 | 30
2      | 5           | 2016-1-3 | 21
3      | 2           | 2016-1-4 | 9
4      | 3           | 2016-3-2 | 10
5      | 2           | 2016-3-1 | 10

为了计算全年的RFM分数,我使用了

now = dt.datetime(2016,12,31)
df.groupby('customer_id').agg({'date': lambda x: (now - x.max()).days,
                               'txn_id': lambda x: len(x),
                               'total : lambda x : sum(x)})
rfm = df['date'] = df['date'].astype(int)
rfm.rename(columns={'date': 'recency', 
                     'txn_id': 'frequency', 
                     'total': 'monetary_value'}, inplace=True)

quantiles = rfm.quantile(q=[0.25,0.5,0.75])
quantiles = quantiles.to_dict()

def RScore(x,p,d):
if x <= d[p][0.25]:
    return 1
elif x <= d[p][0.50]:
    return 2
elif x <= d[p][0.75]: 
    return 3
else:
    return 4

def FMScore(x,p,d):
    if x <= d[p][0.25]:
        return 4
    elif x <= d[p][0.50]:
        return 3
    elif x <= d[p][0.75]: 
        return 2
    else:
        return 1



rfm['R_Quartile'] = rfm['recency'].apply(RClass, args=('recency',quantiles,))
rfm['F_Quartile'] = rfm['frequency'].apply(FMClass, args=('frequency',quantiles,))
rfm['M_Quartile'] = rfm['monetary_value'].apply(FMClass, args=('monetary_value',quantiles,))


rfm['RFMClass'] = rfm.R_Quartile.map(str) \
                            + rfm.F_Quartile.map(str) \
                            + rfm.M_Quartile.map(str)

现在我拥有的是这样的数据框:

 customer_id | RFM
  2          | 313 
  5          | 131   
  3          | 414

我想要按月细分RFM分数,例如:

 customer_id | Jan | Feb | ....| Dec 
  2          | 313 | 324 | ....| 121
  5          | 131 | 342 | ....| 212   
  3          | 414 | 113 | ....| 333

现在对我来说,问题是我不知道如何将所有这些计算都基于月,我考虑过将年循环12次,但是循环太大了,有没有有效的方法吗?

1 个答案:

答案 0 :(得分:0)

我还没有完全按照您的示例进行介绍,但是我相信这可以解决问题。

首先,请确保您的日期实际上是datetime格式(如果您还没有的话)。

data['date'] = pd.to_datetime(data['date'])

然后,创建一个包含月份的新列。

data['month_id'] = data['date'].dt.strftime('%B')

然后,您可以分组并堆叠以获得矩阵,其中矩阵的列为month_id,而行的矩阵为customer_id。这是一个仅包含一个聚合的示例。

data.groupby(['customer_id', 'month_id'])['total'].sum().unstack()

给予:

month_id January March
customer_id     
2        39.0    10.0
3        NaN     10.0
5        21.0    NaN

从那里,您应该能够将RFM分数应用于每一列。请注意,由于您具有多个聚合,因此实际上在数据帧上将具有multiIndex。但是原理是一样的。