Filtro Matlab() con SciPy lfilter()

Question

Secondo la loro documentazione per Matlab filter() e SciPy lfilter(), sembra che dovrebbero essere "compatibili". Tuttavia ho un problema, il porting di codice Matlab più grande in Python, per il quale ottengo ValueError: object of too small depth for desired array. Siccome non riesco a pensare a come posso presentare la mia fonte senza complicare esso, userò l'esempio fornito nella documentazione di Matlab:

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

che traduco in Python per:

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) 

In questo caso ottengo:
ValueError: object too deep for desired array

Perché ottengo questi errori?

Answer 1

C'è una ragione per cui stai aggiungendo una dimensione extra quando crei il tuo array di quelli? È ciò di cui hai bisogno:

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