基于 bin 创建列

Question

我有一个数据：

# dt
Column1     
      1
      2
      3
      4
      5
      6
      7
      8
      9

我想通过 bin 的最小值和最大值的平均值创建一个新列。

# dt
Column1    Column2
      1          2
      2          2
      3          2
      4          5
      5          5
      6          5
      7          8
      8          8
      9          8

pd.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)

或将transform 与add 和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