假设我们有一个包含id、name和age列的数据表,我们想要按id分组,并获取在age列中对应最小值的所有行。以下是一个Python代码示例来解决这个问题:
import pandas as pd
# 创建示例数据表
data = {'id': [1, 1, 2, 2, 3, 3],
'name': ['John', 'Adam', 'Emily', 'Sophia', 'Mike', 'Emma'],
'age': [25, 30, 28, 35, 27, 32]}
df = pd.DataFrame(data)
# 按id分组,并获取在age列中对应最小值的所有行
result = df.loc[df.groupby('id')['age'].idxmin()]
print(result)
运行上述代码,将会得到以下输出:
id name age
0 1 John 25
2 2 Emily 28
4 3 Mike 27
这样,我们成功按id分组,并获取了在age列中对应最小值的所有行。