1
0
mirror of https://github.com/TREX-CoE/qmc-lttc.git synced 2024-12-12 15:33:52 +01:00
qmc-lttc/plot_hydrogen.py
2021-01-25 23:52:53 +01:00

16 lines
319 B
Python

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