如何使用Pandas split来保留列的两部分？

import pandas as pd

# 创建一个示例DataFrame
data = {'column': ['A_B', 'C_D', 'E_F']}
df = pd.DataFrame(data)

# 使用split方法并设置expand=True来保留两部分
df[['part1', 'part2']] = df['column'].str.split('_', expand=True)

print(df)

  column part1 part2
0    A_B     A     B
1    C_D     C     D
2    E_F     E     F