Python基于简单移动平均类

时间:2014-02-18 05:32:31

标签: python class methods average self

好的,所以我正在编写一个可以在价格清单上计算简单移动平均线的课程。它计算每N个价格的平均值,而不计算前N-1天。这就是我所拥有的:

class Simplemovingaverage():
    def __init__(self, Nday, list_of_prices):
        self._Nday = Nday
        self._list_of_prices = list_of_prices

    def calculate(self):
        for i in range(len(self._list_of_prices)):
            if i < self._Nday:
                average = 0
            elif i == self._Nday:
                average = sum(self._list_of_prices[:self._Nday])/self._Nday
            else:
                average = sum(self._list_of_prices[i-self._Nday:i])/self._Nday
            print(average)

我通过在shell 'x = Simplemovingaverage(3, [1,2,3,4,5,6,7,8,9,10])'上创建一个类对象然后通过'x.calculate'执行计算方法来测试它,我得到的输出是:

0
0
0
2.0
3.0
4.0
5.0
6.0
7.0
8.0

所以从我的数字列表中,它的唯一计算结果是7,8,9,最后一个数字应该是9,因为这是8,9,10的平均值,并且由于N是3,所以应该只有3个零。这是我正在寻找的输出:

0
0
2.0
3.0
4.0
5.0
6.0
7.0
8.0
9.0

5 个答案:

答案 0 :(得分:3)

def sma_calc(prices_list, window_size):
    return sum(prices_list[-window_size:]) / window_size

答案 1 :(得分:1)

from __future__ import division
from itertools import islice, tee

def moving_average(n, iterable):
    # leading 0s
    for i in range(1, n):
        yield 0.

    # actual averages
    head, tail = tee(iterable)
    sum_ = float(sum(islice(head, n)))
    while True:
        yield sum_ / n
        sum_ += next(head) - next(tail)

运行时
list(moving_average(3, [1,2,3,4,5,6,7,8,9,10]))

返回

[0.0, 0.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0]

(从N-1前导0开始,意味着输出列表与输入列表具有相同的基数,我认为这是你真正想要的)。

答案 2 :(得分:0)

你有一个错误。试试这个:

  for i in range(len(self._list_of_prices) + 1):

所以这可能会说明发生了什么

>>> _list_of_prices = [1,2,3,4,5,6,7,8,9,10]
>>> len(_list_of_prices)
10
>>> range(len(_list_of_prices))
[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
>>> i = 9
>>> _list_of_prices[i-3:i]
[7, 8, 9]

所以问题是python的列表切片操作符在上面的例子中返回到i而不是i,(在这个例子中它返回 i-3,i-2和i-1)例如。

>>> word = "helpa"
>>> word[0:2]
'he'

请参阅此处http://docs.python.org/2/tutorial/introduction.html

答案 3 :(得分:0)

你误解了索引:

In [186]: class Simplemovingaverage():
     ...:     def __init__(self, Nday, list_of_prices):
     ...:         self._Nday = Nday
     ...:         self._list_of_prices = list_of_prices
     ...: 
     ...:     def calculate(self):
     ...:         for i in range(len(self._list_of_prices)):
     ...:             if i < self._Nday-1:
              #------------------------^^--------------------------
     ...:                 average = 0
     ...:             elif i == self._Nday-1:
              #---------------------------^^--------------------------
     ...:                 average = sum(self._list_of_prices[:self._Nday])/self._Nday
     ...:             else:
     ...:                 average = sum(self._list_of_prices[\
                                i+1-self._Nday:i+1])/self._Nday
              #-----------------^^^-------------^^^--------------------------                  

     ...:             print(average)

In [187]: x = Simplemovingaverage(3, [1,2,3,4,5,6,7,8,9,10])

In [188]: x.calculate()
0
0
2
3
4
5
6
7
8
9

答案 4 :(得分:0)

好的,这是一个清理:

class Simplemovingaverage():
    def __init__(self, navg, items):
        self.navg = navg
        self.items = items

    def calculate(self):
        av = []
        for i in range(len(self.items)):
            if i+1 < self.navg:
                av.append(0)
            else:
                av.append(sum(self.items[i+1-self.navg:i+1])/self.navg)
        return av

首先,您需要在任何地方使用i+1,因为range提供0...navg-1而不是1...navg(您也可以range(1, n+1))。

其次,您不需要特殊情况i+1==self.navg:m0:m相同)。

第三,返回列表而不是打印结果更有意义(尽管我喜欢其他受访者使用yield将其作为生成器的想法!)

第四,没有真正的理由隐藏数字和列表,所以我删除了下划线(python不是java或c ++!)。

最后,这比一定数量的“天”平均“价格列表”更为通用,因此我更改了参数以获得更多的一般性。