Matlab过滤器()与SciPy lfilter()

时间:2012-01-19 07:52:04

标签: python matlab scipy

根据Matlab filter()SciPy lfilter()的文档,它们似乎应该“兼容”。但是我有一个问题,在Python中移植更大的Matlab代码,我得到ValueError: object of too small depth for desired array。由于我无法想象如何在不使其复杂化的情况下呈现我的源代码,我将使用Matlab文档中提供的示例:

data = [1:0.2:4]';
windowSize = 5;
filter(ones(1,windowSize)/windowSize,1,data)

我在Python中翻译为:

import numpy as np
from scipy.signal import lfilter

data = np.arange(1, 4.1, 0.2)
windowSize = 5
lfilter(np.ones((1, windowSize)) / windowSize, 1, data)

在这种情况下,我得到:
ValueError: object too deep for desired array

为什么我会收到这些错误?

1 个答案:

答案 0 :(得分:4)

是否有理由在创建数组时添加额外的维度?这就是你需要的:

lfilter(np.ones(windowSize) / windowSize, 1, data)