Numpy gaussian smoothing

gauss_smooth.py

# Super simple 1D smoothing with just numpy. Just smooths a few sharp edges.

import math
import numpy as np

def gaussian_kernel(n=5):
  x = np.arange(n)/n
  g = math.e ** -(x**2/((n - 2)**2)) # drop the constant factor at the start
  return g/g.sum() # normalise so it sums to 1

def gaussian_smooth(x, n=5, passes=2):
  for x in range(passes):
    x = np.convolve(x, gaussian_kernel(n))
  return x