aszerW
ec749314bc
- 新增使用Tushare获取A股ETF价格及净值数据的私有方法 - fetch_all方法支持接收完整代码配置，区分指数与ETF及市场类别 - 指数数据和ETF数据分别下载，ETF净值数据用于溢价率计算 - 采用A股交易日为主交易日历，非A股数据前向填充对齐 - 调整因子计算，支持指数价格计算因子，ETF价格计算收益率 - run_rotation脚本和RotationStrategy引擎适配指数-ETF配置格式 - 代码结构优化，增强多市场及加密货币处理能力
154 lines
4.2 KiB
Python
154 lines
4.2 KiB
Python
"""
|
||
动量因子计算模块
|
||
|
||
支持两种动量因子：
|
||
1. N日涨幅（简单动量）
|
||
2. 斜率×R²趋势得分（改进版）
|
||
"""
|
||
|
||
import numpy as np
|
||
import pandas as pd
|
||
from sklearn.linear_model import LinearRegression
|
||
|
||
|
||
def calculate_momentum(price_series: pd.Series, n: int) -> pd.Series:
|
||
"""
|
||
计算 N 日涨幅（简单动量）
|
||
|
||
Args:
|
||
price_series: 价格序列
|
||
n: 动量窗口天数
|
||
|
||
Returns:
|
||
Series: N日涨幅
|
||
"""
|
||
return price_series / price_series.shift(n + 1) - 1.0
|
||
|
||
|
||
def _slope_r2_score(srs: pd.Series, n: int = 25) -> float:
|
||
"""
|
||
单次计算斜率×R²趋势得分
|
||
|
||
Args:
|
||
srs: 价格窗口序列（长度为 n）
|
||
n: 窗口长度
|
||
|
||
Returns:
|
||
float: 斜率 × R² × 10000
|
||
"""
|
||
if srs.shape[0] < n:
|
||
return np.nan
|
||
|
||
x = np.arange(1, n + 1).reshape(-1, 1)
|
||
y = srs.values / srs.values[0] # 归一化
|
||
|
||
lr = LinearRegression().fit(x, y)
|
||
slope = lr.coef_[0]
|
||
r_squared = lr.score(x, y)
|
||
score = 10000 * slope * r_squared
|
||
|
||
return score
|
||
|
||
|
||
def calculate_slope_r2(price_series: pd.Series, n: int = 25) -> pd.Series:
|
||
"""
|
||
计算斜率×R²趋势得分序列
|
||
|
||
Args:
|
||
price_series: 价格序列
|
||
n: 滚动窗口天数
|
||
|
||
Returns:
|
||
Series: 趋势得分序列
|
||
"""
|
||
return price_series.rolling(n).apply(
|
||
lambda x: _slope_r2_score(x, n), raw=False
|
||
)
|
||
|
||
|
||
def calculate_daily_return(price_series: pd.Series) -> pd.Series:
|
||
"""
|
||
计算日收益率
|
||
|
||
Args:
|
||
price_series: 价格序列
|
||
|
||
Returns:
|
||
Series: 日收益率
|
||
"""
|
||
return price_series / price_series.shift(1) - 1
|
||
|
||
|
||
def compute_factors(
|
||
index_data: pd.DataFrame,
|
||
code_list: list,
|
||
n: int = 25,
|
||
factor_type: str = "slope_r2",
|
||
etf_data: pd.DataFrame = None,
|
||
code_config: dict = None,
|
||
) -> tuple[pd.DataFrame, list]:
|
||
"""
|
||
计算所有指数的因子和日收益率（支持指数-ETF双轨数据）
|
||
|
||
Args:
|
||
index_data: 指数价格数据（宽格式，用于因子计算）
|
||
code_list: 指数代码列表
|
||
n: 动量/趋势窗口
|
||
factor_type: 'momentum' 或 'slope_r2'
|
||
etf_data: ETF价格数据（宽格式，用于收益计算）
|
||
code_config: 代码配置字典 {code: {name, etf, market}}，用于判断是否为加密货币
|
||
|
||
Returns:
|
||
tuple: (result_df, valid_codes)
|
||
- result_df: 包含因子得分和日收益率的DataFrame
|
||
- valid_codes: 有效代码列表
|
||
"""
|
||
code_config = code_config or {}
|
||
|
||
# 如果没有提供ETF数据，创建一个空的DataFrame
|
||
if etf_data is None:
|
||
etf_data = pd.DataFrame()
|
||
|
||
result = index_data.copy()
|
||
|
||
# 过滤掉缺失值过多的指数
|
||
total_rows = len(result)
|
||
valid_codes = []
|
||
for code in code_list:
|
||
if code not in result.columns:
|
||
print(f" ⚠ 跳过 {code}: 不在数据中")
|
||
continue
|
||
null_pct = result[code].isnull().sum() / total_rows
|
||
if null_pct > 0.2:
|
||
print(f" ⚠ 剔除 {code}: 缺失率 {null_pct:.1%} 过高")
|
||
result = result.drop(columns=[code])
|
||
else:
|
||
valid_codes.append(code)
|
||
|
||
# 对有效指数计算因子和收益率
|
||
for code in valid_codes:
|
||
# 因子基于指数价格计算
|
||
if factor_type == "momentum":
|
||
result[f"得分_{code}"] = calculate_momentum(result[code], n)
|
||
elif factor_type == "slope_r2":
|
||
result[f"得分_{code}"] = calculate_slope_r2(result[code], n)
|
||
else:
|
||
raise ValueError(f"不支持的因子类型: {factor_type}")
|
||
|
||
# 日收益率基于指数价格计算（回测使用指数价格）
|
||
result[f"日收益率_{code}"] = calculate_daily_return(result[code])
|
||
|
||
# 按得分列做 dropna
|
||
score_cols = [f"得分_{code}" for code in valid_codes]
|
||
result = result.dropna(subset=score_cols)
|
||
|
||
print("\n因子计算完成:")
|
||
print(f" 因子类型: {factor_type}")
|
||
print(f" 窗口天数: {n}")
|
||
print(f" 有效指数: {len(valid_codes)}/{len(code_list)}")
|
||
print(f" 有效数据: {len(result)} 行")
|
||
if etf_data is not index_data:
|
||
print(f" 使用ETF数据计算收益: ✓")
|
||
|
||
return result, valid_codes
|