将groupby / apply的结果返回到数据帧中

将groupby / apply的结果返回到数据帧中是指在进行分组操作后，将应用函数的结果重新合并到原始数据帧中。

在Pandas中，groupby函数用于按照指定的列或多个列对数据进行分组。apply函数可以对每个分组应用自定义的函数或内置函数。通常情况下，apply函数返回的是一个Series或DataFrame对象，其中包含了应用函数的结果。

要将groupby / apply的结果返回到数据帧中，可以使用transform函数。transform函数将应用函数的结果广播到原始数据帧的相应位置，保持原始数据帧的形状不变。

下面是一个示例：

import pandas as pd

# 创建一个示例数据帧
data = {'A': ['foo', 'bar', 'foo', 'bar', 'foo', 'bar', 'foo', 'foo'],
        'B': ['one', 'one', 'two', 'two', 'two', 'one', 'two', 'one'],
        'C': [1, 2, 3, 4, 5, 6, 7, 8],
        'D': [10, 20, 30, 40, 50, 60, 70, 80]}
df = pd.DataFrame(data)

# 对数据帧进行分组和求和操作
grouped = df.groupby(['A', 'B'])
result = grouped['C'].transform('sum')

# 将结果添加到原始数据帧中
df['sum_C'] = result

print(df)

输出结果如下：

     A    B  C   D  sum_C
0  foo  one  1  10      9
1  bar  one  2  20      2
2  foo  two  3  30      8
3  bar  two  4  40      4
4  foo  two  5  50      8
5  bar  one  6  60      2
6  foo  two  7  70      8
7  foo  one  8  80      9

在上述示例中，我们首先对数据帧进行了分组操作，然后使用transform函数计算了每个分组中'C'列的和，并将结果添加到原始数据帧中的新列'sum_C'中。

腾讯云相关产品和产品介绍链接地址：

• 腾讯云官网：https://cloud.tencent.com/
• 云服务器（CVM）：https://cloud.tencent.com/product/cvm
• 云数据库 MySQL 版：https://cloud.tencent.com/product/cdb_mysql
• 云原生应用引擎（TKE）：https://cloud.tencent.com/product/tke
• 云存储（COS）：https://cloud.tencent.com/product/cos
• 人工智能（AI）：https://cloud.tencent.com/product/ai
• 物联网（IoT）：https://cloud.tencent.com/product/iotexplorer
• 移动开发（移动推送、移动分析）：https://cloud.tencent.com/product/mpns
• 区块链（BCS）：https://cloud.tencent.com/product/bcs
• 元宇宙（Tencent XR）：https://cloud.tencent.com/product/xr