Как наиболее эффективный способ получить длину пути от матрицы смежности с помощью Numpy? ⇐ Python

[Anonymous]

Проблема, которую я решаю, заключается в оптимизации генетического алгоритма для проблемы с продавцом. Расчет пути занимает больше всего времени. Вот текущий код, над которым я работаю: < /p>
from itertools import pairwise
import numpy as np
from random import shuffle

def get_path_len(adj_mat: np.ndarray, path: np.ndarray) -> float:
return sum(adj_mat[i, j] for i, j in pairwise(path)) + adj_mat[path[-1], path[0]]

mat = np.random.randint(1, 1000, (100, 100))
path = np.asarray(list(range(100)))
shuffle(path)

print(get_path_len(mat, path))


Подробнее здесь: https://stackoverflow.com/questions/795 ... trix-using