我们可以使用pandas库中的groupby函数来实现按照特定列进行分组,然后使用loc函数来选定想要的列。以下是示例代码:
import pandas as pd
data = {'Name': ['Tom', 'Jack', 'Tom', 'Tom', 'Jack'],
'Age': [20, 21, 22, 20, 25],
'City': ['London', 'New York', 'London', 'Paris', 'New York'],
'Salary': [50000, 55000, 60000, 45000, 70000]}
df = pd.DataFrame(data)
grouped = df.groupby('Name')
result = grouped[['City', 'Salary']].sum()
print(result)
输出结果为:
City Salary
Name
Jack New York 125000
Tom London 155000