Python相当于Matlab的持久性

时间:2012-10-06 22:01:53

标签: python decorator

  

可能重复:
  What is the Python equivalent of static variables inside a function?

我尝试编写递归函数。它迭代一个向量并给出一个取决于当前值和先前值的值。在Matlab中,我可以在函数内部将变量声明为persistent,并在调用函数后将值存储在内存中,因此函数的下一次调用将从之前的值开始。

这是我开始的简单移动平均线:

def AvgFilter(x):
    if not firstRun:  # checks if runs for first time, i.e. firstRun is empty
        k = 1        # setup initial variables if run for first time
        prevAvg = 0  # prevAvg - the average calculated during last call
        firstRun = 1 # only for initialisation
    alpha = (k-1)/k  
    avg = alpha * prevAvg + (1 - alpha)*x
    prevAvg = avg
    k = k + 1    
    return avg

我需要在函数调用之间记住变量k prevAvg firstRun。我读过它可以通过装饰器完成,我确实尝试在函数之前实现它@counter,但不确定我应该如何实现它。装饰者是唯一的方式(没有找到别的东西)?以及如何编写计数器函数来存储我的变量?我有点担心以后会有更复杂的递归,我会完全迷失在装饰者的想法中。

3 个答案:

答案 0 :(得分:5)

听起来像generators的工作!生成器让你假装你只是在循环中一次计算一个值,但它实际上会暂停执行并在你调用yield时返回一个值。

def AvgFilter():
    k = 1
    avg = 0
    while True:
        alpha = (k-1)/float(k)
        x = yield avg # yield/return the old average, and get the new input value
        avg = alpha * avg + (1 - alpha)*x
        k = k + 1

f = AvgFilter()
print f.next()
print f.send(1)
print f.send(2)
print f.send(20)
print f.send(20)

# 0
# 1.0
# 1.5
# 7.66666666667
# 10.75

答案 1 :(得分:2)

您可以执行以下操作:

  • 使用对象并将这些变量存储在对象中,在函数调用之间重用相同的对象。
  • 使用全局变量(强烈建议不要!)
  • 使用发电机(参见jtbandes的回答)

以下是一个例子:

class AvgFilter(object):
    def __init__(self):
        self.k = 1
        self.avg = 0

    def run(x):
        alpha = float(self.k-1) / self.k  
        self.avg = alpha * self.avg + (1 - alpha)*x
        self.k +=1
        return self.avg

答案 2 :(得分:2)

感谢@Thomas Orozco和@jtbandes的回答经过几天的漫长的一天,我得到了我想要的工作。代码和一些测试是:

x = [10,2,30,4,50]
class avgFil2(object):
    def __init__(self):
        self.avg = 0
        self.k = 1.0
    def __call__(self, x):
        alpha = (self.k-1.0)/self.k  
        self.avg = alpha * self.avg + (1 - alpha)*x
        self.k +=1
       return self.avg

def simpleTest(x = x):     
    average = []
    avg = avgFil2()               # new variable and creation of instance (all initialisation)
    for i in range(len(x)):
        print 'input %f' % x[i] 
        print 'before change avg.avg=%f, k=%f' % (avg.avg, avg.k)   
        average.append(avg(x[i])) # class is called here, so all changes going on
        print 'after change avg.avg=%f, k=%f' % (avg.avg, avg.k)  
        print 'The output average is %f' % average[i]

simpleTest()

代码与Matlab完全不同,但很不错。重要的陷阱是将k=1.0初始化为float而非k=1初始化为int,因为alpha将为零,因为除法将评估为整数。另一件事,我必须初始化为'self.k'而不是'k',因为它给出了错误。生成器可能也可以正常工作,但我喜欢class解决方案。