以下是一个示例代码,可以按照列进行分组,并显示每个组的IDs:
import pandas as pd
# 创建一个示例数据集
data = {'ID': [1, 2, 3, 4, 5, 6],
'Group': ['A', 'B', 'A', 'B', 'A', 'B']}
df = pd.DataFrame(data)
# 按照 'Group' 列进行分组,并显示每个组的IDs
grouped = df.groupby('Group')['ID'].apply(list)
print(grouped)
输出结果:
Group
A [1, 3, 5]
B [2, 4, 6]
Name: ID, dtype: object
这个示例中,我们使用了pandas库来处理数据。首先,我们创建了一个示例数据集,并将其存储在一个DataFrame中。然后,使用groupby
方法按照'Group'列进行分组。最后,使用apply
方法将每个组的'ID'列转换为列表,并将结果存储在一个Series中。