基于 bin 创建列

Question

我有一个数据：# dtColumn1           1      2      3      4      5      6      7      8      9我想通过 bin 的最小值和最大值的平均值创建一个新列。# dtColumn1    Column2      1          2      2          2      3          2      4          5      5          5      6          5      7          8      8          8      9          8pd.qcut(dt['Column1'], 3)所以 column2 = (bin 的最小值 + bin 的最大值)/2。

Answer 1

GroupBy.transform与 lambda 函数Series一起使用以返回与原始大小相同的大小DataFrame：

dt['Column2'] = (dt.groupby(pd.qcut(dt['Column1'], 3))['Column1']

                   .transform(lambda x: x.max() + x.min()) / 2)

或者使用 doubletransform和addand div：

g = dt.groupby(pd.qcut(dt['Column1'], 3))

dt['Column2'] = g['Column1'].transform('max').add(g['Column1'].transform('min')).div(2)

print (dt)

   Column1  Column2

0        1      2.0

1        2      2.0

2        3      2.0

3        4      5.0

4        5      5.0

5        6      5.0

6        7      8.0

7        8      8.0

8        9      8.0

编辑：

cols = ['Column1']

for col in cols:

    dt[f'New {col}'] = (dt.groupby(pd.qcut(dt[col], 3))[col]

                       .transform(lambda x: x.max() + x.min()) / 2)

print (dt)

   Column1  New Column1

0        1          2.0

1        2          2.0

2        3          2.0

3        4          5.0

4        5          5.0

5        6          5.0

6        7          8.0

7        8          8.0

8        9          8.0