#!/usr/bin/python3
import matplotlib.pyplot as plt
import numpy as np
from phys import *

quantum_system = box(0.001) # box with inf walls

# x = np.arange(0, BOX_LENGTH, 0.00001)
# n = np.arange(1, 10, 1)
# 
# X, N = np.meshgrid(x, n)
# P = prob(X, N)
# 
# fig = plt.figure()
# ax = plt.axes(projection='3d')
# ax.plot_surface(X, N, P, rstride=1, cstride=1, cmap='viridis', edgecolor='none')
# ax.set_xlabel('x [m]')
# ax.set_ylabel('n [int]')
# ax.set_zlabel('probability [frac]');
# 
# plt.show()