在Python中,可以使用pandas
库来实现按组进行分组并选择每组的第一个、第二个和第四个成员。以下是一个示例代码:
import pandas as pd
# 创建一个示例数据集
data = {'Group': ['A', 'A', 'A', 'B', 'B', 'B', 'C', 'C', 'C'],
'Member': ['M1', 'M2', 'M3', 'M1', 'M2', 'M3', 'M1', 'M2', 'M3']}
df = pd.DataFrame(data)
# 按照'Group'列进行分组,并选择每组的第一个、第二个和第四个成员
result = df.groupby('Group').nth([0, 1, 3])
print(result)
输出结果为:
Member
Group
A M1
A M2
B M1
B M2
C M1
C M2
上述代码中,首先使用pd.DataFrame()
函数创建一个包含两列('Group'和'Member')的数据集。然后,使用groupby()
函数按照'Group'列进行分组。最后,使用nth()
函数选择每组的第一个、第二个和第四个成员(索引0、1和3)。