如何将Pandas中的月份和年列合并为yyyymm格式？

df['月份'] = df['月份'].astype(str)
df['年'] = df['年'].astype(str)

df['yyyymm'] = df.apply(lambda row: row['年'] + row['月份'], axis=1)

import pandas as pd

# 创建示例数据
data = {'年': [2020, 2021, 2022],
        '月份': [1, 2, 3]}
df = pd.DataFrame(data)

# 将月份和年列合并为yyyymm格式
df['月份'] = df['月份'].astype(str)
df['年'] = df['年'].astype(str)
df['yyyymm'] = df.apply(lambda row: row['年'] + row['月份'], axis=1)

print(df)

      年  月份  yyyymm
0  2020   1  20201
1  2021   2  20212
2  2022   3  20223