如何在Panda中对一列中的值进行分组，并在另一列中获取相应的值？

grouped = df.groupby('A')

result = grouped['B'].apply(lambda x: x.values[0])

import pandas as pd

# 创建示例数据
data = {'A': ['group1', 'group1', 'group2', 'group1', 'group2'],
        'B': [1, 2, 3, 4, 5]}
df = pd.DataFrame(data)

# 按照列A的值进行分组
grouped = df.groupby('A')

# 在分组后的数据中获取列B的值
result = grouped['B'].apply(lambda x: x.values[0])

print(result)

A
group1    1
group2    3
Name: B, dtype: int64