下面是一个使用Python实现的示例代码,通过pandas库按组分组,并使用自定义函数处理每组的前4行:
import pandas as pd
# 创建示例数据
data = {'Group': ['A', 'A', 'A', 'A', 'B', 'B', 'B', 'B'],
'Value': [1, 2, 3, 4, 5, 6, 7, 8]}
df = pd.DataFrame(data)
# 自定义函数处理每组的前4行
def custom_function(group):
return group.iloc[:4, :]
# 按组分组,并使用自定义函数处理每组的前4行
result = df.groupby('Group').apply(custom_function)
# 打印结果
print(result)
运行以上代码,将会得到如下输出:
Group Value
Group
A 0 A 1
1 A 2
2 A 3
3 A 4
B 4 B 5
5 B 6
6 B 7
7 B 8
以上代码首先创建了一个包含两列(Group和Value)的DataFrame,然后定义了一个自定义函数custom_function
,该函数用于处理每个组的前4行。最后,通过groupby
方法按组分组,并使用apply
方法应用自定义函数处理每组的前4行。结果被存储在一个新的DataFrame中,并打印输出。
上一篇:按组分组,转换-恢复到原始表
下一篇:按组复制控制行