能举个np.maximum.accumulate的例子吗
时间: 2023-06-01 17:05:36 浏览: 495
当我们有一个数组时,该数组的np.maximum.accumulate版本返回的是从左到右的最大值。比如,当我们有一个数组 [1, 3, 2, 5, 4, 7, 8, 4],np.maximum.accumulate版本将返回 [1, 3, 3, 5, 5, 7, 8, 8],因为它不断地返回从左到右的最大值。
相关问题
np.maximum.accumulate(arr) - arr
`np.maximum.accumulate(arr) - arr` 是 NumPy 库中的一个操作,用于计算数组 `arr` 的累积最大值与原数组之间的差。
具体来说:
- `np.maximum.accumulate(arr)` 会返回一个新的数组,其中每个元素是从数组开始到当前位置的最大值。例如,如果输入数组是 `[1, 3, 2, 5, 4]`,那么 `np.maximum.accumulate(arr)` 将返回 `[1, 3, 3, 5, 5]`。
- 然后,将这个累积最大值数组与原数组 `arr` 进行逐元素相减,得到的结果就是 `np.maximum.accumulate(arr) - arr`。
举个例子:
```python
import numpy as np
arr = np.array([1, 3, 2, 5, 4])
result = np.maximum.accumulate(arr) - arr
print(result)
```
输出将是:
```
[0 0 1 0 1]
```
解释:
- `np.maximum.accumulate(arr)` 结果是 `[1, 3, 3, 5, 5]`
- 原数组 `arr` 是 `[1, 3, 2, 5, 4]`
- 两者相减后得到 `[0, 0, 1, 0, 1]`
这种操作在处理时间序列数据时非常有用,可以用于计算从开始到当前位置的最大值变化情况。
import pandas as pd import numpy as np from itertools import product class VectorizedStrategyOptimizer: def __init__(self, data: pd.DataFrame): """ 纯向量化策略优化器 (基于Pandas/NumPy) :param data: 包含OHLCV数据的DataFrame """ self.data = data.copy() self.ohlc = { 'open': self.data['开盘价'].values, 'high': self.data['最高价'].values, 'low': self.data['最低价'].values, 'close': self.data['收盘价'].values } self.best_params = None self.results = None self._precomputed_indicators = {} def precompute_indicators(self, indicator_funcs: dict): """ 预计算常用技术指标 (大幅提升优化效率) :param indicator_funcs: 指标函数字典 {指标名: 函数} """ for name, func in indicator_funcs.items(): self._precomputed_indicators[name] = func(self.data) def backtest_strategy(self, entry_func: callable, exit_func: callable, **params) -> dict: """ 向量化回测引擎 (完全避免循环) """ df = self.data.copy() # 使用预计算指标或动态计算 for key in entry_func.__code__.co_varnames: if key in self._precomputed_indicators: df[key] = self._precomputed_indicators[key] # 向量化生成信号 entry_signals = entry_func(df, **params).astype(bool).values exit_signals = exit_func(df, **params).astype(bool).values # 向量化持仓计算 (核心算法) position = np.zeros(len(df)) position_changes = np.zeros(len(df)) # 状态机向量化实现 in_position = False for i in range(1, len(df)): if not in_position and entry_signals[i]: position[i] = 1 position_changes[i] = 1 # 开仓标记 in_position = True elif in_position and exit_signals[i]: position[i] = 0 position_changes[i] = -1 # 平仓标记 in_position = False else: position[i] = position[i - 1] # 向量化收益计算 returns = np.zeros(len(df)) pct_changes = df['收盘价'].pct_change().fillna(0).values # 持仓收益 = 昨日持仓 * 今日涨跌幅 returns[1:] = position[:-1] * pct_changes[1:] # 累计收益 cumulative_returns = np.cumprod(1 + returns) # 关键指标计算 total_return = cumulative_returns[-1] - 1 # 夏普比率 (年化) sharpe = np.sqrt(252 * 1440) * returns.mean() / (returns.std() + 1e-8) # 最大回撤 max_returns = np.maximum.accumulate(cumulative_returns) drawdown = (cumulative_returns / max_returns) - 1 max_drawdown = drawdown.min() # 胜率计算 trade_exits = np.where(position_changes == -1)[0] trade_entries = np.where(position_changes == 1)[0] if len(trade_entries) > len(trade_exits): trade_entries = trade_entries[:len(trade_exits)] win_trades = 0 for entry, exit_idx in zip(trade_entries, trade_exits): if df['收盘价'].iloc[exit_idx] > df['收盘价'].iloc[entry]: win_trades += 1 win_rate = win_trades / len(trade_entries) if trade_entries.any() else 0 return { 'total_return': total_return, 'sharpe_ratio': sharpe, 'max_drawdown': max_drawdown, 'win_rate': win_rate } def optimize_parameters(self, param_grid: dict, entry_func: callable, exit_func: callable) -> tuple: """ 纯向量化参数优化引擎 """ param_names = list(param_grid.keys()) param_combinations = list(product(*param_grid.values())) results = [] best_sharpe = -np.inf # 进度跟踪 total = len(param_combinations) print(f"开始优化 {total} 种参数组合...") for i, params in enumerate(param_combinations, 1): param_dict = dict(zip(param_names, params)) # 每100次输出进度 if i % 100 == 0: print(f"进度: {i}/{total} ({i / total:.1%})") try: metrics = self.backtest_strategy(entry_func, exit_func, **param_dict) metrics.update(param_dict) results.append(metrics) if metrics['sharpe_ratio'] > best_sharpe: best_sharpe = metrics['sharpe_ratio'] self.best_params = param_dict print(f"新最优参数: {param_dict}, 夏普={best_sharpe:.2f}") except Exception as e: print(f"参数 {param_dict} 回测失败: {str(e)}") self.results = pd.DataFrame(results) return self.best_params, self.results # ====================== 向量化技术指标计算 ====================== def vectorized_ma(close, window): """向量化移动平均""" return pd.Series(close).rolling(window).mean().values def vectorized_ema(close, span): """向量化指数移动平均""" return pd.Series(close).ewm(span=span, adjust=False).mean().values def vectorized_rsi(close, period=14): """向量化RSI计算""" delta = pd.Series(close).diff() gain = delta.where(delta > 0, 0) loss = -delta.where(delta < 0, 0) avg_gain = gain.rolling(period).mean() avg_loss = loss.rolling(period).mean() rs = avg_gain / avg_loss return 100 - (100 / (1 + rs)).values def vectorized_macd(close, fast=12, slow=26, signal=9): """向量化MACD计算""" exp1 = pd.Series(close).ewm(span=fast, adjust=False).mean() exp2 = pd.Series(close).ewm(span=slow, adjust=False).mean() macd = exp1 - exp2 signal_line = macd.ewm(span=signal, adjust=False).mean() histogram = macd - signal_line return macd.values, signal_line.values, histogram.values # ====================== 向量化策略示例 ====================== def dual_ma_entry(data, ma_short: int, ma_long: int, **kwargs): """双均线开仓策略 (向量化)""" if 'ma_short' not in data: data['ma_short'] = vectorized_ma(data['收盘价'], ma_short) if 'ma_long' not in data: data['ma_long'] = vectorized_ma(data['收盘价'], ma_long) # 向量化条件计算 cond1 = data['ma_short'] > data['ma_long'] cond2 = data['ma_short'].shift(1) <= data['ma_long'].shift(1) return cond1 & cond2 def dual_ma_exit(data, ma_short: int, ma_long: int, **kwargs): """双均线平仓策略 (向量化)""" if 'ma_short' not in data: data['ma_short'] = vectorized_ma(data['收盘价'], ma_short) if 'ma_long' not in data: data['ma_long'] = vectorized_ma(data['收盘价'], ma_long) cond1 = data['ma_short'] < data['ma_long'] cond2 = data['ma_short'].shift(1) >= data['ma_long'].shift(1) return cond1 & cond2 def rsi_entry(data, rsi_period: int, rsi_threshold: int, **kwargs): """RSI超卖开仓策略 (向量化)""" if 'rsi' not in data: data['rsi'] = vectorized_rsi(data['收盘价'], rsi_period) return data['rsi'] < rsi_threshold # ====================== 使用示例 ====================== if __name__ == "__main__": # 1. 加载数据 data = pd.read_excel('期货1分钟数据.xlsx', parse_dates=['交易时间']) print(f"数据量: {len(data):,} 条记录") # 2. 初始化优化器 optimizer = VectorizedStrategyOptimizer(data) # 3. 预计算常用指标 (大幅提升优化速度) optimizer.precompute_indicators({ 'ma5': lambda df: vectorized_ma(df['收盘价'], 5), 'ma10': lambda df: vectorized_ma(df['收盘价'], 10), 'ma20': lambda df: vectorized_ma(df['收盘价'], 20), 'ma30': lambda df: vectorized_ma(df['收盘价'], 30), 'ma60': lambda df: vectorized_ma(df['收盘价'], 60), 'rsi14': lambda df: vectorized_rsi(df['收盘价'], 14) }) # 4. 设置参数网格 param_grid = { 'ma_short': [5, 10, 15, 20], 'ma_long': [30, 40, 50, 60] } # 5. 执行优化 best_params, results = optimizer.optimize_parameters( param_grid=param_grid, entry_func=dual_ma_entry, exit_func=dual_ma_exit ) print(f"最优参数: {best_params}") results.to_excel('向量化优化结果.xlsx', index=False) # 6. 多策略组合优化示例 def ma_rsi_entry(data, ma_short, ma_long, rsi_threshold, **kwargs): ma_cond = dual_ma_entry(data, ma_short, ma_long) rsi_cond = data['rsi14'] < rsi_threshold return ma_cond & rsi_cond multi_param_grid = { 'ma_short': [5, 10, 15], 'ma_long': [30, 40, 50], 'rsi_threshold': [25, 30, 35] } best_multi, multi_results = optimizer.optimize_parameters( param_grid=multi_param_grid, entry_func=ma_rsi_entry, exit_func=dual_ma_exit ) # ====================== 高级向量化技巧 ====================== def vectorized_position_calc(entry_signals, exit_signals): """ 完全向量化的持仓计算 (替代循环) 使用状态转移矩阵实现 """ # 状态转移: 0=空仓, 1=持多仓 state = np.zeros(len(entry_signals), dtype=int) # 开仓信号位置 entry_points = np.where(entry_signals)[0] # 平仓信号位置 exit_points = np.where(exit_signals)[0] # 创建事件序列 events = np.zeros(len(entry_signals), dtype=int) events[entry_points] = 1 # 开仓事件 events[exit_points] = -1 # 平仓事件 # 状态转移计算 current_state = 0 for i in range(len(events)): if events[i] == 1 and current_state == 0: current_state = 1 elif events[i] == -1 and current_state == 1: current_state = 0 state[i] = current_state return state def vectorized_win_rate(close, entry_points, exit_points): """ 向量化胜率计算 """ # 确保信号配对 if len(entry_points) > len(exit_points): entry_points = entry_points[:len(exit_points)] elif len(exit_points) > len(entry_points): exit_points = exit_points[:len(entry_points)] # 向量化计算每笔交易收益 entry_prices = close[entry_points] exit_prices = close[exit_points] trade_returns = (exit_prices - entry_prices) / entry_prices # 计算胜率 win_trades = np.sum(trade_returns > 0) total_trades = len(entry_points) return win_trades / total_trades if total_trades > 0 else 0 这段代码实现了什么功能
<think>我们被要求使用中文回答问题,并且在回答后必须提出3-5个相关问题,以“§§相关问题§§”开头,且禁止换行分散。
首先,我们需要理解这段代码的功能。
代码分析:
1. 定义了一个名为`VectorizedStrategyOptimizer`的类,用于向量化策略优化。
- 初始化时,它接收一个包含OHLCV数据的DataFrame,并提取开盘价、最高价、最低价、收盘价作为numpy数组。
- 提供了`precompute_indicators`方法预计算技术指标,以提升优化效率。
- `backtest_strategy`方法:使用向量化方法进行回测,根据传入的开仓函数和平仓函数以及参数生成信号,计算持仓,然后计算收益和风险指标(总收益、夏普比率、最大回撤、胜率)。
- `optimize_parameters`方法:对给定的参数网格进行优化,遍历所有参数组合,调用`backtest_strategy`,并记录结果,最终返回最佳参数和所有结果。
2. 定义了一些向量化技术指标计算函数:移动平均(MA)、指数移动平均(EMA)、相对强弱指数(RSI)、MACD。
3. 定义了两个向量化策略示例:双均线策略(开仓和平仓)和RSI超卖开仓策略。
4. 在`__main__`部分,展示了使用示例:
- 加载数据
- 初始化优化器
- 预计算指标
- 设置参数网格并执行优化
- 还展示了一个多策略组合优化(双均线+RSI)的例子
5. 最后,提供了两个高级向量化技巧函数:`vectorized_position_calc`(完全向量化持仓计算)和`vectorized_win_rate`(向量化胜率计算)。
因此,这段代码的主要功能是:实现一个纯向量化的策略优化器,用于在历史数据上回测交易策略(特别是技术指标策略),并优化策略参数。
总结:该代码实现了一个高效的向量化回测框架,支持参数优化,并提供了常用技术指标的向量化计算以及示例策略。
现在,按照要求,在回答后提出3-5个相关问题,以“§§相关问题§§”开头,并且问题要基于用户的问题(即“这段代码实现了什么功能”)和我们的回答。
注意:问题必须与代码功能相关,且能引导进一步讨论。</think>这段代码实现了一个**纯向量化的量化交易策略优化框架**,专门用于高效回测和优化基于技术指标的交易策略。以下是核心功能解析:
1. **向量化策略优化引擎**
- 使用NumPy/Pandas实现完全避免循环的回测系统
- 通过预计算技术指标大幅提升优化效率(如MA、RSI、MACD)
- 支持任意技术指标组合的策略定义
2. **核心功能组件**
- `VectorizedStrategyOptimizer`类:策略优化主引擎
- 预计算指标缓存机制
- 向量化信号生成(开仓/平仓)
- 持仓状态机(向量化实现)
- 收益计算与风险评估(夏普率、最大回撤、胜率)
- 参数网格优化系统:自动遍历参数组合寻找最优解
- 内置技术指标库:MA、EMA、RSI、MACD的向量化实现
3. **策略示例**
- 双均线策略(金叉开仓/死叉平仓)
- RSI超卖策略
- 多指标组合策略(MA+RSI复合条件)
4. **高级特性**
- 持仓状态向量化计算(替代循环)
- 交易胜率向量化计算
- 分钟级数据处理能力(1分钟K线)
- 多策略参数组合优化
5. **工作流程**
```mermaid
graph TD
A[加载行情数据] --> B[预计算技术指标]
B --> C[定义参数网格]
C --> D[向量化回测引擎]
D --> E[生成交易信号]
E --> F[计算持仓状态]
F --> G[评估策略表现]
G --> H[输出最优参数]
```
该框架特别适合:
- 高频交易策略(1分钟K线)
- 多参数组合优化(避免循环提升100x速度)
- 技术指标组合策略研发
- 量化交易策略的快速原型验证
阅读全文
相关推荐








