Как я могу векторизировать функцию, которая возвращает собственные значения и собственные векторы матрицы в Python? ⇐ Python

[Anonymous]

Я работаю с функцией в Python, которая строит матрицу 4 × 4 на основе входов (x1, Y1, x2, y2), и вычисляет ее собственные значения и собственные векторы, использующие np.linalg.eigh
PrettyPrint-Override ">

Код: Выделить всё

import numpy as np

def f(kx, ky):
return kx + 1j * ky

def fs(kx, ky):
return np.conj(f(kx, ky))

def eig(x1, y1, x2, y2):
a = 10
x = x1 + x2
y = y1 + y2
H = np.array([
[a, f(x, y), f(x, y), fs(x, y)],
[fs(x, y), a, 0, f(x, y)],
[fs(x, y), 0, -a, f(x, y)],
[f(x, y), fs(x, y), fs(x, y), -a]
])

Eval, Evec = np.linalg.eigh(H)
sorted_indices = np.argsort(Eval)
return Eval[sorted_indices], Evec[:, sorted_indices]

< /code>
Теперь у меня есть 1-D массивы входных значений: < /p>
x1_array, y1_array, x2_array, y2_array  # all same shape

Я хочу эффективно векторизировать эту функцию в этих массивах - то есть вычислить все собственные значения/собственные векторы.>

Подробнее здесь: https://stackoverflow.com/questions/796 ... rs-of-a-ma