动量因子策略全解析：捕捉趋势的阿尔法来源

import pandas as pd
import numpy as np

def calculate_momentum(price_data, lookback=12, lag=1):
    """
    计算动量因子
    
    参数:
    - price_data: 价格数据 (DataFrame, index为日期, columns为股票代码)
    - lookback: 回溯期（月）, 默认12个月
    - lag: 滞后期（月）, 默认1个月
    
    返回:
    - momentum_score: 动量得分 (DataFrame)
    """
    # 转换为月度数据
    monthly_prices = price_data.resample('M').last()
    
    # 计算动量得分 (过去12个月收益率，排除最近1个月)
    momentum_score = monthly_prices.pct_change(periods=lookback) - \
                     monthly_prices.pct_change(periods=lag)
    
    return momentum_score

import pandas as pd
import numpy as np
from backtrader import Cerebro, Strategy, DataFeed
import matplotlib.pyplot as plt

class MomentumStrategy(Strategy):
    """
    动量因子策略
    """
    params = (
        ('lookback', 12),  # 回溯期（月）
        ('lag', 1),        # 滞后期（月）
        ('top_n', 20),    # 持有Top N只股票
        ('rebalance_freq', 'M'),  # 调仓频率
    )
    
    def __init__(self):
        self.momentum_score = {}
        self.rebalance_date = None
        
    def next(self):
        # 检查是否需要调仓
        current_date = self.datas[0].datetime.date(0)
        if self.rebalance_date is None or \
           current_date >= self.rebalance_date:
            
            self.rebalance()
            
            # 设置下次调仓日期
            if self.params.rebalance_freq == 'M':
                # 下个月第一天
                next_month = pd.Timestamp(current_date).replace(day=1) + pd.DateOffset(months=1)
                self.rebalance_date = next_month.date()
    
    def rebalance(self):
        # 计算所有股票的动量得分
        momentum_scores = []
        for data in self.datas:
            if len(data) >= self.params.lookback * 20:  # 假设每月20个交易日
                returns = data.close.get(size=-self.params.lookback*20)
                momentum = (returns[-1] / returns[0]) - 1
                momentum_scores.append((data._name, momentum))
        
        # 按动量得分排序
        momentum_scores.sort(key=lambda x: x[1], reverse=True)
        
        # 选择Top N只股票
        target_stocks = [x[0] for x in momentum_scores[:self.params.top_n]]
        
        # 调仓：卖出不在目标列表中的股票，买入目标股票
        # ... (具体调仓逻辑)