在 PyAlgoTrade 中使用多种工具进行回测

Question

您好，我想使用一个数组 ("instruments") 将 1 到 10 种可能投资的策略概括为简化加载 10 个提要、创建 10 个 SMA 的任务，然后每天检查是否一个（或多个）仪器发生信号交叉。

我被困在了这里。绘图仪也在单独绘制图表，但我想将所有仪器结果绘制在一张图表中。

这是我的代码：

from pyalgotrade import strategy, plotter
from pyalgotrade.barfeed import yahoofeed
from pyalgotrade.technical import ma
from pyalgotrade.tools import yahoofinance

class MyStrategy(strategy.BacktestingStrategy):
    def __init__(self, feed, instruments, smaPeriod):
        strategy.BacktestingStrategy.__init__(self, feed, 1000)
        self.__position = None
        # We'll use adjusted close values instead of regular close values.
        self.setUseAdjustedValues(True)
        self.__sma = {}
        self.__instruments = instruments
        for instrument in instruments:
            self.__sma[instrument] = ma.SMA(feed[instrument].getPriceDataSeries(), smaPeriod)

    def onEnterOk(self, position):
        execInfo = position.getEntryOrder().getExecutionInfo()
        self.info("BUY at $%.2f" % (execInfo.getPrice()))

    def onEnterCanceled(self, position):
        execInfo = position.getEntryOrder().getExecutionInfo()

    def onExitOk(self, position):
        execInfo = position.getExitOrder().getExecutionInfo()
        self.info("SELL at $%.2f" % (execInfo.getPrice()))

    def onExitCanceled(self, position):
        # If the exit was canceled, re-submit it.
        self.__position[str(position.getEntryOrder().getInstrument())].exitMarket()

    def onBars(self, bars):
        # Wait for enough bars to be available to calculate a SMA.
        if self.__sma[-1] is None:
            return

        bar = bars[self.__instrument]
        # If a position was not opened, check if we should enter a long position.
        if self.__position is None:
            if bar.getPrice() > self.__sma[-1]:
            # Enter a buy market order for 25 shares. The order is good till canceled.
                self.__position = self.enterLong(self.__instrument, 25, True)
        # Check if we have to exit the position.
        elif bar.getPrice() < self.__sma[-1] and not self.__position.exitActive():
            self.__position.exitMarket()

def run_strategy(smaPeriod):

    # Load the yahoo feed from the CSV file
    instruments = [
        "AMZN",
        "ADBE",
        "C" ,
        "BA" ,
        "HOG" ,
        "MMM" ,
        "MS" ,
        "MSFT" ,
        "CVS" ,
        "AXP"
    ]
    #Download and Load yahoo feed from CSV files
    #Change year range 2000 to 2001 to your desired one
    feed = yahoofinance.build_feed(instruments, 2000,2001, ".")

    # Evaluate the strategy with the feed.
    myStrategy = MyStrategy(feed, instruments, smaPeriod)

    # Attach a plotter to the strategy
    plt = plotter.StrategyPlotter(myStrategy)


    # Run the strategy
    myStrategy.run()
    print "Final portfolio value: $%.2f" % myStrategy.getBroker().getEquity()

    # Plot the strategy.
    plt.plot()


run_strategy(10)

Answer 1

我刚刚开始处理 pyalgotrade，但我认为您犯了一个相当简单的错误（如 gzc 所示）：class Bars 的一个实例是来自具有相同时间戳的不同工具。因此，当您的 onBars 事件被调用时，您实际上必须遍历字典中的所有乐器。

在 PyAlgoTrade 中使用多种工具进行回测

Backtesting using multiple instruments in PyAlgoTrade

python

pyalgotrade