把柱子和Pandas关在一起

rseugnpd  于 2023-01-28  发布在  其他
关注(0)|答案(4)|浏览(112)

我有一个包含数值的数据框列:

df['percentage'].head()
46.5
44.2
100.0
42.12

我希望将列显示为bin counts

bins = [0, 1, 5, 10, 25, 50, 100]

如何获得具有 * 值计数 * 的柱形结果?

[0, 1] bin amount
[1, 5] etc
[5, 10] etc
...
c90pui9n

c90pui9n1#

您可以使用pandas.cut

bins = [0, 1, 5, 10, 25, 50, 100]
df['binned'] = pd.cut(df['percentage'], bins)
print (df)
   percentage     binned
0       46.50   (25, 50]
1       44.20   (25, 50]
2      100.00  (50, 100]
3       42.12   (25, 50]
bins = [0, 1, 5, 10, 25, 50, 100]
labels = [1,2,3,4,5,6]
df['binned'] = pd.cut(df['percentage'], bins=bins, labels=labels)
print (df)
   percentage binned
0       46.50      5
1       44.20      5
2      100.00      6
3       42.12      5

或者numpy.searchsorted

bins = [0, 1, 5, 10, 25, 50, 100]
df['binned'] = np.searchsorted(bins, df['percentage'].values)
print (df)
   percentage  binned
0       46.50       5
1       44.20       5
2      100.00       6
3       42.12       5

......然后是value_countsgroupby,并聚合size
一个三个三个一个
默认情况下,cut返回categorical
Series方法(如Series.value_counts())将使用所有类别,即使某些类别不存在于数据中,也是如此。

8gsdolmq

8gsdolmq2#

使用Numba模块加速。

在大数据集(大于500k)上,pd.cut对数据进行分箱可能会非常慢。
我用Numba编写了自己的函数,使用了即时编译,速度大约快了 * 6倍 *:

from numba import njit

@njit
def cut(arr):
    bins = np.empty(arr.shape[0])
    for idx, x in enumerate(arr):
        if (x >= 0) & (x < 1):
            bins[idx] = 1
        elif (x >= 1) & (x < 5):
            bins[idx] = 2
        elif (x >= 5) & (x < 10):
            bins[idx] = 3
        elif (x >= 10) & (x < 25):
            bins[idx] = 4
        elif (x >= 25) & (x < 50):
            bins[idx] = 5
        elif (x >= 50) & (x < 100):
            bins[idx] = 6
        else:
            bins[idx] = 7

    return bins
cut(df['percentage'].to_numpy())

# array([5., 5., 7., 5.])

可选:您也可以将其作为字符串Map到bin:

a = cut(df['percentage'].to_numpy())

conversion_dict = {1: 'bin1',
                   2: 'bin2',
                   3: 'bin3',
                   4: 'bin4',
                   5: 'bin5',
                   6: 'bin6',
                   7: 'bin7'}

bins = list(map(conversion_dict.get, a))

# ['bin5', 'bin5', 'bin7', 'bin5']
    • 速度比较**:

一个一个三个一个一个一个一个一个四个一个一个一个一个一个五个一个

puruo6ea

puruo6ea3#

我们也可以使用np.select

bins = [0, 1, 5, 10, 25, 50, 100]
df['groups'] = (np.select([df['percentage'].between(i, j, inclusive='right') 
                           for i,j in zip(bins, bins[1:])], 
                          [1, 2, 3, 4, 5, 6]))

输出:

percentage  groups
0       46.50       5
1       44.20       5
2      100.00       6
3       42.12       5
7eumitmz

7eumitmz4#

使用Numpy的方便快捷版本

np.digitize是一个方便快捷的选项:

import pandas as pd
import numpy as np

df = pd.DataFrame({'x': [1,2,3,4,5]})
df['y'] = np.digitize(a['x'], bins=[3,5])

print(df)

退货

x  y
0  1  0
1  2  0
2  3  1
3  4  1
4  5  2

相关问题