如何根据同一公司同一数据框另一列的往年值修改数据框中的列值

import pandas as pd

# 创建一个示例数据框
data = {
    '公司': ['A', 'A', 'B', 'B', 'C', 'C'],
    '年份': [2020, 2021, 2020, 2021, 2020, 2021],
    '销售额': [100, 120, 200, 220, 300, 320]
}
df = pd.DataFrame(data)

# 根据同一公司同一数据框另一列的往年值修改数据框中的列值
df['去年销售额'] = df.groupby('公司')['销售额'].shift(1)

# 输出修改后的数据框
print(df)

  公司    年份  销售额  去年销售额
0  A  2020  100    NaN
1  A  2021  120  100.0
2  B  2020  200    NaN
3  B  2021  220  200.0
4  C  2020  300    NaN
5  C  2021  320  300.0