以下是一个示例代码,其中按列分组并找到两个组中的ID:
import pandas as pd
# 创建示例数据
data = {
'Group': ['A', 'A', 'B', 'B', 'B', 'A'],
'ID': [1, 2, 3, 4, 5, 6]
}
df = pd.DataFrame(data)
# 按列分组并找到两个组中的ID
grouped = df.groupby('Group')
for name, group in grouped:
if len(group) == 2:
print(f"Group {name}: {group['ID'].tolist()}")
输出结果:
Group A: [1, 2]
Group B: [3, 4]
在示例代码中,我们首先创建了一个包含两列(Group和ID)的示例数据框。然后,我们使用groupby()
函数按Group列分组。
接下来,我们使用for
循环迭代每个分组,并使用len()
函数检查每个分组的大小。如果分组大小为2,则打印该组的ID列。