在Pandas中，在两个单独的列中返回第一个值，然后返回所有其他值

import pandas as pd

# 创建示例数据
data = {'A': ['A1', 'A1', 'A2', 'A2', 'A3', 'A3'],
        'B': ['B1', 'B2', 'B3', 'B4', 'B5', 'B6'],
        'C': ['C1', 'C2', 'C3', 'C4', 'C5', 'C6']}
df = pd.DataFrame(data)

# 在两个单独的列中返回第一个值
df['First_A'] = df.groupby('A')['A'].transform('first')
df['First_B'] = df.groupby('B')['B'].transform('first')

# 返回所有其他值
df['Other_A'] = df['A'][df['A'] != df['First_A']]
df['Other_B'] = df['B'][df['B'] != df['First_B']]

print(df)

    A   B   C First_A First_B Other_A Other_B
0  A1  B1  C1      A1      B1     NaN     B2
1  A1  B2  C2      A1      B1      A1     NaN
2  A2  B3  C3      A2      B3     NaN     B4
3  A2  B4  C4      A2      B3      A2     NaN
4  A3  B5  C5      A3      B5     NaN     B6
5  A3  B6  C6      A3      B5      A3     NaN