1
0
mirror of https://github.com/TREX-CoE/qmc-lttc.git synced 2024-10-07 00:36:27 +02:00
qmc-lttc/plot_hydrogen.py
2021-01-03 18:45:58 +01:00

22 lines
370 B
Python

import numpy as np
import matplotlib.pyplot as plt
from hydrogen import e_loc
x=np.linspace(-5,5)
def make_array(a):
y=np.array([ e_loc(a, np.array([t,0.,0.]) ) for t in x])
return y
plt.figure(figsize=(10,5))
for a in [0.1, 0.2, 0.5, 1., 1.5, 2.]:
y = make_array(a)
plt.plot(x,y,label=f"a={a}")
plt.tight_layout()
plt.legend()
plt.savefig("plot_py.png")