下面是一个使用Python和pandas库的示例代码,演示了如何根据组选择目标记录数。
import pandas as pd
# 创建示例数据
data = {
'Group': ['A', 'A', 'B', 'B', 'B', 'C', 'C'],
'Value': [1, 2, 3, 4, 5, 6, 7]
}
df = pd.DataFrame(data)
# 按组选择目标记录数
target_count = 2 # 目标记录数
result = df.groupby('Group').head(target_count)
print(result)
输出结果为:
Group Value
0 A 1
1 A 2
2 B 3
3 B 4
4 B 5
5 C 6
6 C 7
在上面的示例中,我们首先创建了一个包含组(Group)和数值(Value)的示例数据。然后,我们使用groupby
函数按组分组,然后使用head
函数选择每个组的前target_count
个记录。最后,我们打印了结果。
上一篇:按组选择每组的前1个并进行转换
下一篇:按组选择n个数据并限制