在分箱列上分组不能正常工作

Question

我正在使用 pandas cut 来计算一个新的 bins 列，如下所示：bins = [1, 10, 20, 34, np.Inf]labels = ['1-10', '11-20', '21-34', '35 -Inf']df['binned'] = pd.cut(df['Number of Locations'], bins=bins, labels=labels, include_lowest=True)这为我提供了合并值的新列，然后我尝试下面的代码：df.groupby(['binned', 'Parent_Account'])['has_desired_product'].apply(sum).reset_index()这应该给我按新的 binned 列分组，但它给了我不正确的输出 - 实际上只有一个 Parent_Account 用于'35-inf' bin 但它显示的不止于此，我的代码某处有错误吗？

Answer 1

没有提供样本数据。测试边缘情况似乎都很好。我正在使用熊猫 1.1

df = pd.DataFrame({"Number of Locations":[32,33,34,35,36,np.inf,np.nan,34.0001]})

bins = [1, 10, 20, 34, np.Inf]

labels = ['1-10', '11-20', '21-34', '35 -Inf']

df['binned'] = pd.cut(df['Number of Locations'], bins=bins, labels=labels, include_lowest=True)

print(df.to_string())

输出

   Number of Locations   binned

0              32.0000    21-34

1              33.0000    21-34

2              34.0000    21-34

3              35.0000  35 -Inf

4              36.0000  35 -Inf

5                  inf  35 -Inf

6                  NaN      NaN

7              34.0001  35 -Inf