函数为所谓的不同输入返回相同的值

时间:2016-11-26 19:45:13

标签: python function pyalgotrade

我正在使用pyalgotrade创建交易策略。我正在查看代码清单(testlist)并将它们添加到字典(list_large {})以及我使用get_score函数获得的分数。我的最新问题是字典中的每个自动收报机(list_large {})获得了相同的分数。知道为什么吗?

代码:

from pyalgotrade import strategy
from pyalgotrade.tools import yahoofinance
import numpy as np
import pandas as pd
from collections import OrderedDict

from pyalgotrade.technical import ma
from talib import MA_Type
import talib

smaPeriod = 10
testlist = ['aapl','ddd','gg','z']

class MyStrategy(strategy.BacktestingStrategy):
    def __init__(self, feed, instrument):
        super(MyStrategy, self).__init__(feed, 1000)
        self.__position = [] 
        self.__instrument = instrument
        self.setUseAdjustedValues(True)
        self.__prices = feed[instrument].getPriceDataSeries()
        self.__sma = ma.SMA(feed[instrument].getPriceDataSeries(), smaPeriod)

    def get_score(self,slope):
        MA_Score = self.__sma[-1] * slope
        return MA_Score

    def onBars(self, bars): 

        global bar 
        bar = bars[self.__instrument]

        slope = 8

        for instrument in bars.getInstruments():

            list_large = {}
            for tickers in testlist: #replace with real list when ready
                list_large.update({tickers : self.get_score(slope)}) 

            organized_list = OrderedDict(sorted(list_large.items(), key=lambda t: -t[1]))#organize the list from highest to lowest score

         print list_large


def run_strategy(inst):
    # Load the yahoo feed from the CSV file

    feed = yahoofinance.build_feed([inst],2015,2016, ".") # feed = yahoofinance.build_feed([inst],2015,2016, ".")

    # Evaluate the strategy with the feed.
    myStrategy = MyStrategy(feed, inst)
    myStrategy.run()
    print "Final portfolio value: $%.2f" % myStrategy.getBroker().getEquity()


def main():
    instruments = ['ddd','msft']
    for inst in instruments:
            run_strategy(inst)


if __name__ == '__main__':
        main()

1 个答案:

答案 0 :(得分:0)

检查onBars()功能的代码:

slope = 8    # <---- value of slope = 8 

for instrument in bars.getInstruments():
    list_large = {}
    for tickers in testlist: #replace with real list when ready
        list_large.update({tickers : self.get_score(slope)}) 
        #       Updating dict of each ticker based on ^

每次调用self.get_score(slope)时,它都会返回相同的值,因此tickers的所有值都在dict中保持相同的值

我不知道您希望如何处理slope以及如何更新它的价值。但是,可以简化此逻辑而不使用.update

list_large = {}
for tickers in testlist: 
    list_large[tickers] = self.get_score(slope)
     #           ^ Update value of `tickers` key