gauss.py

'''
A few gaussian curves
'''

import matplotlib.pyplot as plt
import numpy as np

def gaussian(a, b, c, d, x):
    '''
    Gaussian function
    '''
    return a*np.exp(-((x-b)**2)/(2*c**2)) + d

print(gaussian(1, 2, 3, 4, 5))


X = np.linspace(-10, 10)

Y = [gaussian(1, -5, 5, 1, x) for x in X]
plt.plot(X, Y)

Y = [gaussian(2, 0, 1, 1, x) for x in X]
plt.plot(X, Y)

Y = [gaussian(2, 2, 1, 2, x) for x in X]
plt.plot(X, Y)

plt.legend()
plt.show()