Skip to content

Instantly share code, notes, and snippets.

@ahurriyetoglu
Forked from bhawkins/medfilt.py
Created August 26, 2014 20:22
Show Gist options
  • Save ahurriyetoglu/b0141916ebf5ed0f958d to your computer and use it in GitHub Desktop.
Save ahurriyetoglu/b0141916ebf5ed0f958d to your computer and use it in GitHub Desktop.
#!/usr/bin/env python
import numpy as np
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)
def test ():
import pylab as p
x = np.linspace (0, 1, 101)
x[3::10] = 1.5
p.plot (x)
p.plot (medfilt (x,3))
p.plot (medfilt (x,5))
p.show ()
if __name__ == '__main__':
test ()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment