human
I am working on the file "Numerical Solution to the 1D Steady State Schrödinger Equation.ipynb".
The first task is:
First, let's define some potential functions:
Can you write Python code for it?

=====
Give me a the Python code you requested?
assistant
#Here is the Python code you requested.

def gaussian(center, height, sigma):
    '''Gaussian Potential'''
    return lambda x : height * np.exp(-(x - center)**2/(2 * sigma**2))

def square(height, l, r):
    '''Square Well Potential'''
    def V(x):
        if l < x < r:
            return height
        else:
            return 0.0
    return V