Матричное возведение в степень в Python
Я пытаюсь построить сложную матрицу в Python и столкнулся с некоторыми проблемами. Я используюscipy.linalg.expm
функция, и я получаю довольно странное сообщение об ошибке, когда я пытаюсь следующий код:
import numpy as np
from scipy import linalg
hamiltonian = np.mat('[1,0,0,0;0,-1,0,0;0,0,-1,0;0,0,0,1]')
# This works
t_list = np.linspace(0,1,10)
unitary = [linalg.expm(-(1j)*t*hamiltonian) for t in t_list]
# This doesn't
t_list = np.linspace(0,10,100)
unitary = [linalg.expm(-(1j)*t*hamiltonian) for t in t_list]
Ошибка при запуске второго эксперимента:
This works!
Traceback (most recent call last):
File "matrix_exp.py", line 11, in <module>
unitary_t = [linalg.expm(-1*t*(1j)*hamiltonian) for t in t_list]
File "/usr/lib/python2.7/dist-packages/scipy/linalg/matfuncs.py", line 105, in expm
return scipy.sparse.linalg.expm(A)
File "/usr/lib/python2.7/dist- packages/scipy/sparse/linalg/matfuncs.py", line 344, in expm
X = _fragment_2_1(X, A, s)
File "/usr/lib/python2.7/dist- packages/scipy/sparse/linalg/matfuncs.py", line 462, in _fragment_2_1
X[k, k] = exp_diag[k]
TypeError: only length-1 arrays can be converted to Python scalars
Это кажется действительно странным, поскольку все, что я изменил, былоt
Я использовал. Это потому, что гамильтониан диагональн? В общем, гамильтонианов не будет, но я также хочу, чтобы он работал для диагональных. Я действительно не знаю механикуexpm
, поэтому любая помощь будет принята с благодарностью.