import nitrogen as n2
import matplotlib.pyplot as plt
import numpy as np

my_dvr = n2.basis.SimpleDVR(start=-1.0, stop=1.0, num=11, basis='ho')
x = np.linspace(-3,3,251) # make a grid for plotting
y = my_dvr.wfs(x)
plt.plot(x,y[:,[0,-1]],'-')
plt.plot(my_dvr.grid, np.zeros_like(my_dvr.grid), 'k.')