In [1]:

    

import sys
import h5py
import numpy as np
from mpl_toolkits.mplot3d import Axes3D
from matplotlib import pyplot as plt


    

In [3]:

    

#/usr/bin/python
# -*- coding: utf-8 -*-
f = h5py.File("res_temp.h5",'r')
dset = f.get("/numres")
dset2 = f.get("/params1d")
p = np.array(dset2)
a = np.array(dset)
f.close()
dset=0;
dset2=0;
nx=int(p[4])
ne = int(p[6])
x=np.linspace(0,p[0],num=int(p[4]))
fig=plt.figure()
ax=fig.add_subplot(111)
print(len(x))
print(len(a))
print(p)
fig = plt.figure()
#ax1 = fig.add_subplot(221)
#ax2 = fig.add_subplot(222)
#ax3 = fig.add_subplot(223)
#ax4 = fig.add_subplot(224)

for i in range(50,100):
    ax.plot(a[i*nx+1:(i+1)*nx])
#for i in range(11,20):
#    ax2.plot(x,a[i*nx:(i+1)*nx])   
#for i in range(21,30):
#    ax3.plot(x[0:nx],a[i*nx:(i+1)*nx])
#for i in range(31,40):
#    ax4.plot(x[0:nx],a[i*nx:(i+1)*nx])
#ax1.set_title("$\psi_n(x)$ for $E_0$-$E_9$")
#ax1.set_xlabel("$x(a.u.)$")
#ax1.set_ylabel("$\psi_n(x)$")       
#ax2.set_title("$\psi_n(x)$ for $E_{10}$-$E_{19}$")
#ax2.set_xlabel("$x(a.u.)$")
#ax2.set_ylabel("$\psi_n(x)$")    
#ax3.set_title("$\psi_n(x)$ for $E_{20}$-$E_{29}$")
#ax3.set_xlabel("$x(a.u.)$")
#ax3.set_ylabel("$\psi_n(x)$")
#ax4.set_title("$\psi_n(x)$ for $E_{30}$-$E_{39}$")
#ax4.set_xlabel("$x(a.u.)$")
#ax4.set_ylabel("$\psi_n(x)$")
   
a=0
x=0   
fig.clf()
plt.show()


    

    




1000000
128000000
[  5.00000000e+01   0.00000000e+00   5.00000000e+00   0.00000000e+00
   1.00000000e+06   2.00000000e+00   1.00000000e+03]

In [ ]:

    




    

In [52]:

    

E=-1.0
nx=1e5
ne=5
dE=E/ne
x=500.0
xmin=-10.0
psi=np.zeros(int(nx))
def V(x,t):
    return 1/np.sqrt(x**2)
psi[1]=-1e-5
t=0
a=np.linspace(0,x,int(nx))
while(t<ne):
    E=t*dE
    print(E)
    for i in range(2,int(nx)):
        dx=(x-xmin)/nx;
        f1=(1+(dx**2)/6*((-E+V(xmin+dx*(i+1.),0.))*2))**(-1.)
        f2=(1-5/6*dx**2*((-E+V(i*dx+xmin,0))*2))
        f3=(1+dx**2/6*(2*(-E+V((i-1)*dx+xmin,0))))
        psi[i] = f1*(2*psi[i-1]*f2-psi[i-2]*f3)   
    pp.plot(a,psi)    
    t+=1
pp.xlabel("x $(a_0)$",size="20")
pp.ylabel("$\Psi(x)$",size="20")
pp.legend()
pp.show()
psi=0


    

    




-0.0
-0.2
-0.4
-0.6000000000000001
-0.8






    




/usr/lib/python3.5/site-packages/matplotlib/axes/_axes.py:519: UserWarning: No labelled objects found. Use label='...' kwarg on individual plots.
  warnings.warn("No labelled objects found. "

In [ ]: