在pandas上向量化函数-未知窗口长度/索引范围

import pandas as pd

# 创建一个示例DataFrame
data = {
    'date': pd.date_range(start='1/1/2020', periods=10),
    'value': [10, 20, 30, 40, 50, 60, 70, 80, 90, 100]
}
df = pd.DataFrame(data)
df.set_index('date', inplace=True)

# 定义一个函数来计算移动平均值，窗口大小根据某些条件动态确定
def dynamic_moving_average(series, window_size):
    return series.rolling(window=window_size).mean()

# 假设窗口大小是根据日期动态确定的
df['moving_avg'] = df['value'].apply(lambda x: dynamic_moving_average(df['value'], window_size=3))

print(df)

# 示例：根据日期动态调整窗口大小
def get_window_size(date):
    # 这里可以根据日期或其他条件来确定窗口大小
    if date.month in [1, 2, 12]:  # 冬季
        return 5
    else:
        return 3

df['moving_avg_dynamic'] = df.index.map(lambda date: dynamic_moving_average(df.loc[:date, 'value'], window_size=get_window_size(date)))

print(df)