python - 在 python 中将数据分类到 bin 的有效方法

Question

假设我有一个 float 据集 (x)，它可以采用 0.0 到 1.0 之间的任何值。我想将数据分类到自定义容器中，例如:

    cat= 0 # the output category
    if x > 0.8 and x<=0.9:
        cat = 1
    if x > 0.7 and x<=0.8:
        cat=2
    if x>0.6 and x<=0.7:
        cat = 3

等等...这是最有效的(就我必须写多少行而言)方法吗？我在想是否有某种方法可以只指定类别的上下范围和类别编号，而不必编写那么多 if 语句。

Answer 1

我建议您将数据移至 pandas 数据框

df['data'] = pd.DataFrame(x)
binInterval = [0, 0.6, 0.7, 0.8, 0.9]
binLabels   = [0, 4, 3, 2, 1]
df['binned'] = pd.cut(df['data'], bins = binInterval, labels=binLabels)

引用文档 here