Векторизация по строкам массива

Question

У меня есть массив X, и я хочу применить функцию f ко всем строкам X:

# silly example
X = numpy.array([[1, 2, 3, 4, 5],
                 [6, 7, 8, 9, 0]], 'i')

def f(row): return sum(row)

y = numpy.vectorize(f, 'i')(rows(X))

Теперь y должно быть array([15,30], 'i').Какой метод или магия нарезки реализует rows наиболее эффективным способом?

Answer 1

NumPy реализует концепцию «действия над определенной осью». Общая функция numpy.apply_along_axis():

>>> numpy.apply_along_axis(sum, 1, X)
array([15, 30])

(где sum, конечно, можно заменить на что угодно).

Answer 2

Вот еще один пример, который учитывает тип и размер результата:

numpy.fromiter((your_func(row) for row in X), dtype=bool, count=len(X))

Даже если цикл не является циклом C, установка типа и размера результата можетпомочь ускорить процесс.

Answer 3

Должно ли это быть что-то, предоставленное NumPy?Потому что я просто вижу понимание списка

[action_to_apply(row) for row in X]