代码计算1D中值滤波器

Question

我想知道是否有人知道一些python或java代码来计算1D中值过滤器。我有一个文件，用两个字段分隔:日期和信号。像这样:

2014-06-01 11:22:12, 23.8

2014-06-01 11:23:12, 25.9

2014-06-01 11:24:12, 45.7

我想阅读这个文件并应用尺寸为23的1D中值过滤器，并将其保存在另一个文件中，以消除噪声。

提前感谢。

亚历山大。

Answer 1

以防以后有人无意中发现。

要提取数据，您可以使用正则表达式，而对于自定义中值过滤器，您可以在这里查看。我将在这里留下一个副本，以防它被删除:

def medfilt (x, k):
    """Apply a length-k median filter to a 1D array x.
    Boundaries are extended by repeating endpoints.
    """
    assert k % 2 == 1, "Median filter length must be odd."
    assert x.ndim == 1, "Input must be one-dimensional."
    k2 = (k - 1) // 2
    y = np.zeros ((len (x), k), dtype=x.dtype)
    y[:,k2] = x
    for i in range (k2):
        j = k2 - i
        y[j:,i] = x[:-j]
        y[:j,i] = x[0]
        y[:-j,-(i+1)] = x[j:]
        y[-j:,-(i+1)] = x[-1]
    return np.median (y, axis=1)

Answer 2

scipy.signal。medfilt接受1D内核:

import pandas as pd
import scipy.signal

def median_filter(file_name, new_file_name, kernel_size):
    with open(file_name, 'r') as f:
        df = pd.read_csv(f, header=None)
    signal = df.iloc[:, 1].values
    median = scipy.signal.medfilt(signal, kernel_size)
    df = df.drop(df.columns[1], 1)
    df[1] = median
    df.to_csv(new_file_name, sep=',', index=None, header=None)

if __name__=='__main__':
    median_filter('old_signal.csv', 'new_signal.csv',  23)