使用python创建RFM模型

RFM是客户分析及衡量客户价值的重要模型之一，其中的R表示最近一次消费(Recency)，F表示消费频率(Frequency)，M表示消费金额(Monetary)。依据这三个指标RFM可以将客户动态分组，并进一步指导客户运营的下一步工作。本篇文章使用python的聚类算法创建一个粗糙的RFM模型。

1. 开始前的准备工作

首先是开始前的准备工作，这里除了常用的numpy和pandas以外，有几个库文件单独说明下，pymysql是python连接mysql进行数据提取的库，cluster用于聚类，此外还有一些库文件在本篇文章中并没有使用到，例如DecisionTreeClassifier，grid_search和cross_validation等等。这些会在后面对RFM模型进行进一步分类预测，模型持久化和依据RFM模型做内容推荐时使用到。因此先一起放在这里，后面的系列文章会进行使用说明。

#导入库文件

import numpy as np

import pandas as pd

import pymysql

from sklearn.cluster import KMeans

from sklearn.tree import DecisionTreeClassifier

import os

from sklearn import cross_validation

import matplotlib.pyplot as plt

from sklearn.externals import joblib

from math import sqrt

from sklearn import grid_search

from sklearn.ensemble import RandomForestClassifier

2. 从mysql提取RFM指标数据

原始的用户数据存储在mysql中，因此需要使用python连接mysql进行数据提取。下面是具体的代码和数据提取过程。

#连接数据库连接

db = pymysql.connect(host=””,user=” “,passwd=” “,port=,db=” ” )

# 使用 cursor() 方法创建一个游标对象 cursor

cursor = db.cursor()

#计算并提取RFM指标的SQL语句

sql=”SELECT a.`user_id`,ABS(TIMESTAMPDIFF(YEAR,NOW(),DATE_FORMAT(MID(b.`id_no`,7,8),’%Y-%m-%d’))) AS age, a.`client_type`,a.`acct_bal`,a.`amount`,TIMESTAMPDIFF(DAY,NOW(),MAX(FROM_UNIXTIME(trans_time,’%y-%m-%d %H:%i:%s’))) AS R,COUNT(id) AS F,SUM(a.`amount`)AS M FROM qb_account_detail AS a JOIN qb_user_info AS b ON a.`user_id`=b.`user_id` WHERE a.`trans_time`> ‘1498838400’ AND a.`trans_type`=’201′ AND NOT ISNULL(id_no) AND id_no <> ‘null’ GROUP BY b.`id_no`;”

# 使用 execute() 方法执行 SQL 查询

cursor.execute(sql)

895

# 使用 fetchall() 方法获取全部数据.

data = cursor.fetchall()

#查看数据

print(data)

使用提取的数据创建DataFrame数据表，取名为rfm。

#创建RFM数据表

columnlist=[‘user_id’,’age’,’client_type’,’acct_bal’,’amount’,’recency’,’frequency’,’monetary’]

rfm=pd.DataFrame.from_records(list(data),columns=columnlist)

查看数据表的前5行数据，这里除了创建RFM模型必要的指标以外，还保留了用户ID，年龄等信息，也是为了后面的分类和推荐使用。