[Python: median filter (1d)] #median #filter #medfilt #window #python

med_filt.py

#!/usr/bin/env python3

import window # https://gist.github.com/estysdesu/af7301fa17817c0d0ac7b5f8952c9bfe

def median_filt_1d(data, w_size):
    """ Median filter for 1d data. Edges are handled with a window shift. """
    windowed_data = window(data, w_size)
    w_half_size = w_size//2
    for i, w_data in enumerate(windowed_data):
        w_data_sort = sorted(w_data)
        data[i] = w_data_sort[w_half_size] # zero indexing handles shift to middle
    return data