AttributeError:'Backtest'对象没有属性'data',熊猫,SMA交叉

问题描述 投票:0回答:1

[尝试使此代码正常工作,但不确定我是否正确传递了父类,使用self.data是否正确?是否可以有一种更简洁的方式来编写此代码...

import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
import pandas_datareader.data as web
import os
import pandas_datareader as pdr

start = pd.to_datetime("Jan 1st 2013")
end = pd.to_datetime("Dec 31st 2018")

class StockData:
    ticker = []
    def stock(self,symbol):
        self.ticker.clear()
        self.ticker.append(symbol)
        self.data= web.DataReader(self.ticker,"yahoo", start, end)

class Backtest(StockData):
    def prepare_indicators(self):
        self.data['SMA1'] = self.data[self.symbol].rolling(self.SMA1).mean()
        self.data['SMA2'] = self.data[self.symbol].rolling(self.SMA2).mean()
        self.data['SMA_1'] = self.data[self.symbol].rolling(self.SMA1).mean().shift(1)
        self.data['SMA_2'] = self.data[self.symbol].rolling(self.SMA2).mean().shift(1)
    def backtest_strategy(self, SMA1, SMA2):
        self.SMA1 = SMA1
        self.SMA2 = SMA2
        self.prepare_indicators()
        print(self.data)
        self.data['Signal'] = np.where((self.data['SMA1'] > self.data['SMA2']) & (self.data['SMA_1'] < self.data['SMA_2']), 1, 0)
        self.data['Signal'] = np.where((self.data['SMA1'] < self.data['SMA2']) & (self.data['SMA_1'] > self.data['SMA_2']), 0, self.data['Signal'])
        self.data['Sell'] = self.data.apply(lambda x : x['Adj Close'] if x['SMA1'] > x['SMA2'] and x['SMA_1'] < x['SMA_2'] else 0, axis=1)
        self.data['Buy'] = self.data.apply(lambda y : y['Adj Close'] if y['SMA1'] < y['SMA2'] and y['SMA_1'] > y['SMA_2'] else 0, axis=1)
        self.data['TP'] = self.data['Buy'] + self.data['Sell']
        self.data['TP'] = self.data['TP'].replace(to_replace=0, method='ffill')
        self.data['Position'] = self.data['Signal'].replace(to_replace=0, method= 'ffill')
        self.data['Buy & Hold Returns'] = np.log(self.data['Adj Close'] / self.data['Adj Close'].shift(1))
        self.data['Strategy Returns'] = self.data['Buy & Hold Returns'] * Self.data['Position'].shift(1)
        self.data[['Buy & Hold Returns', 'Strategy Returns']].cumsum().plot(grid=True, figsize=(9,5))

sma = Backtest()


sma.backtest_strategy(50,200)

AttributeError:“回测”对象没有属性“数据”

python pandas moving-average algorithmic-trading
1个回答
0
投票

如何在类中定义变量的模式如下:

class Xxxx:
    staticVar = ...

    def __init__(self):
        self.instanceVar = ...

所以应该定义类中使用的每个变量:

  • 作为static变量(不是由以下任何特定实例“拥有”此类),在任何方法之外在此类中创建,
  • 或作为instance变量(由特定实例拥有),由构造函数。

一个类变量也可以在该类内的其他方法中定义。

您的代码包含一个分配了[[self.data的地方,在StockData.stock方法中,但此方法被never调用。

您引用

self.data的其他地方的形式为:

self.data[...] = self.data[...]...
因此,此变量必须更早创建。

一种可能的解决方案是添加一个构造函数(也许到

StockData

)并将此变量设置为所需的任何值。或者也许您应该在代码中的某个地方调用

stock

另一个概念是:

    将此函数转换为构造函数(保留
  • symbol
参数),当您调用
  • sma = Backtest()
  • 时,为此参数传递一些值。但是可能您应该将

    append

    更改为extend。原因是如果您将字符串作为symbol传递,则它实际上也是一个字符列表,因此ticker将包含单个字符
    。但是,如果更改为extend,则整个字符串将被添加为一个元素。
    © www.soinside.com 2019 - 2024. All rights reserved.