Оценить выражение массивов numpy по набору индексов ⇐ Python

[Anonymous]

Я хочу вычислить выражение, состоящее из массивов numpy, но только по заданному индексу. Например, предположим, что у меня есть массивы numpy a, b и c и я хочу вычислить (a + b[j]) * c[k, i]. Я мог бы сделать что-то вроде

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

total = 0
for (i, j, k) in indices:
total += (a[i] * b[j]) + c[k, i]

но я ищу наиболее оптимизированный способ сделать это. Есть ли способ сделать это в чистом виде, чтобы избежать цикла for? Я ищу общее решение, которое будет работать для произвольных выражений, включающих + и *.

Подробнее здесь: https://stackoverflow.com/questions/793 ... of-indices