Я пытаюсь построить сложную матрицу в 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, поэтому любая помощь будет принята с благодарностью.