Применить функцию ко всем элементам в массиве - PullRequest
0 голосов
/ 25 февраля 2019

У меня есть некоторый массив:

array = np.array([[1, 0], [1, 0], [1, 0]])

Я хочу вычислить np.kron таким образом:

def func(array):
    res = array[0]
    for i in range(1, len(array)):
        res = np.kron(res, array[i]) 
    return res

Но как это сделать вболее тупой способ?

Я пытался np.apply_along_axis :

res = np.apply_along_axis(np.kron, 1, array)

Но у меня ничего не получилось.

1 Ответ

0 голосов
/ 25 февраля 2019

Я не знаю, является ли это более "тупым", но определенно более питоническим способом было бы использовать встроенную функцию уменьшения:

reduce(np.kron, array)
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...