我在 Jupyter 中运行了这个并将 exampledata.txt 放在与笔记本相同的目录中。
请注意第一行:
df = pd.DataFrame(datas, columns=['userid', 'recency', 'frequency', 'monetary'])
当未在数据文件中定义时加载列 'userid'。我删除了这个列名。
解决方案
import pandas as pd
def pct_rank_qcut(series, n):
edges = pd.Series([float(i) / n for i in range(n + 1)])
f = lambda x: (edges >= x).argmax()
return series.rank(pct=1).apply(f)
datas = pd.read_csv('./exampledata.txt', delimiter=';')
df = pd.DataFrame(datas, columns=['recency', 'frequency', 'monetary'])
df['recency'] = df['recency'].astype(float)
df['frequency'] = df['frequency'].astype(float)
df['monetary'] = df['monetary'].astype(float)
df['recency'] = pct_rank_qcut(df.recency, 5)
df['frequency'] = pct_rank_qcut(df.frequency, 5)
df['monetary'] = pct_rank_qcut(df.monetary, 5)
说明
您看到的问题是 pd.qcut 假设 5 个大小相同的 bin 的结果。在您提供的数据中,'frequency' 有超过 28% 的数字 1。这打破了qcut。
我提供了一个新函数 pct_rank_qcut 来解决这个问题并将所有 1 推入第一个 bin。
edges = pd.Series([float(i) / n for i in range(n + 1)])
此行根据n 定义的所需箱数定义了一系列百分位边缘。在n = 5 的情况下,边缘将为[0.0, 0.2, 0.4, 0.6, 0.8, 1.0]
f = lambda x: (edges >= x).argmax()
这一行定义了一个辅助函数以应用于下一行的另一个系列。 edges >= x 将返回一个长度等于edges 的系列,其中每个元素是True 或False,具体取决于x 是否小于或等于该边。对于x = 0.14,生成的(edges >= x) 将是[False, True, True, True, True, True]。通过采用argmax(),我确定了系列为True 的第一个索引,在本例中为1。
return series.rank(pct=1).apply(f)
这一行接受输入series 并将其转换为百分位排名。我可以将这些排名与我创建的边缘进行比较,这就是我使用apply(f) 的原因。返回的应该是一系列编号为 1 到 n 的 bin 编号。这一系列 bin 编号与您尝试使用的相同:
pd.qcut(df['recency'].values, 5).codes + 1
这会导致 bin 不再相等,并且 bin 1 完全从 bin 2 借用。但必须做出一些选择。如果您不喜欢这个选择,请使用这个概念来建立您自己的排名。
演示
print df.head()
recency frequency monetary
0 3 5 5
1 2 5 5
2 2 5 5
3 1 5 5
4 2 5 5
更新
pd.Series.argmax() 现已弃用。只需切换到pd.Series.values.argmax()() 即可更新!
def pct_rank_qcut(series, n):
edges = pd.Series([float(i) / n for i in range(n + 1)])
f = lambda x: (edges >= x).values.argmax()
return series.rank(pct=1).apply(f)