fork download
  1. import numpy as np
  2. from numpy import linalg as la
  3. N = 4
  4. qmax = 5
  5. qmin = - qmax
  6. dq = (qmax - qmin)/(N-1)
  7. epslion = 0.5/dq/dq
  8. M = np.zeros((N,N))
  9. U = np.zeros(N)
  10. q = np.zeros(N)
  11. for val in range(0,N):
  12. q[val] = qmin + dq * val
  13. U[val] = 0.5 * q[val] * q[val]
  14.  
  15. M[0][0] = 2 *epslion + U[0]
  16. M[0][1] = - epslion
  17. forval in range(1,N-1):
  18. M[val][val-1] = - epslion
  19. M[val][val] = 2 * epslion + U[val]
  20. M[val][val+1] = - epslion
  21.  
  22. M[N-1][N-2] = - epslion
  23. M[N-1][N-1] = 2 * epslion + U[N-1]
  24.  
  25. eg,vt = la.eigh(M)
  26. print("eigenvalue")
  27. print(eg)
  28. for val in range(0,N):
  29. print(q[val],vt.T[0,val],vt.T[1,val],vt.T[2,val],vt.T[3,val])
Runtime error #stdin #stdout #stderr 0.01s 28384KB
stdin
Standard input is empty
stdout
Standard output is empty
stderr
  File "./prog.py", line 17
    forval in range(1,N-1):
                          ^
SyntaxError: invalid syntax