1
0
mirror of https://github.com/TREX-CoE/qmc-lttc.git synced 2024-12-11 06:53:54 +01:00
qmc-lttc/plot_hydrogen.py

16 lines
319 B
Python
Raw Normal View History

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