如何用另一列的滚动均值创建新列-- Python

import pandas as pd

# 创建一个示例数据集
data = {'A': [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]}
df = pd.DataFrame(data)

# 计算滚动均值并创建新列
window_size = 3  # 滚动窗口大小
df['滚动均值'] = df['A'].rolling(window_size).mean()

# 打印结果
print(df)

    A      滚动均值
0   1       NaN
1   2       NaN
2   3  2.000000
3   4  3.000000
4   5  4.000000
5   6  5.000000
6   7  6.000000
7   8  7.000000
8   9  8.000000
9  10  9.000000