从回归到分类:将浮动转换为类别

Question

我有一个数组，其中包含一个人员列表的年龄：[41, 50, 12, ... , 94]

我想将这些映射到20类别。有没有一种方法可以自动检测列表中的最佳类别(例如[15,20,22,24,28,.., 90]？(，然后在其上映射现有阵列。

Answer 1

您可以使用pandas剪切函数，它会将年龄映射到正确的bin：

import pandas as pd
ages = [41, 50, 12, 78,43, 94]
bin_edges = [0,10,15,20,30,50,70,100]
pd.cut(ages,bin_edges)