Как применить scipy.signal.filtfilt () к неполным данным

Question

Я хочу построить неполные данные (некоторые значения - Нет). Кроме того, я хочу применить функцию масла к набору данных и показать оба графика, неполные и сглаженные. Кажется, что функция фильтра не работает с неполными данными.

Файл данных: data.csv

import matplotlib.pyplot as plt
import pandas as pd
import numpy as np
from scipy import signal

data = np.genfromtxt('data.csv', delimiter = ',')
df = pd.DataFrame(data)
df.set_index(0, inplace = True)

b, a = signal.butter(5, 0.1)
y = signal.filtfilt(b,a, df[1].values)
df2 = pd.DataFrame(y, index=df.index)

df.plot()
df2.plot()

plt.show()

Answer 1

На странице документации ничего не сказано о NaN. Возможно, вам придется сначала удалить NaN из списка значений. Вот способ сделать это с помощью функции Numpy isnan:

y = signal.filtfilt(b, a, df[1].values[~np.isnan(df[1].values)])
df2 = pd.DataFrame(y, index=df.index[~np.isnan(df[1].values)])