python - 迭代 Numpy 矩阵行以分别应用一个函数？

Question

我希望能够遍历矩阵以将函数应用于每一行。我怎样才能为 Numpy 矩阵做到这一点？

Answer 1

您可以使用 numpy.apply_along_axis()。假设你的数组是二维的，你可以像这样使用它:

import numpy as np

myarray = np.array([[11, 12, 13],
                    [21, 22, 23],
                    [31, 32, 33]])
def myfunction(x):
    return x[0] + x[1]**2 + x[2]**3

print(np.apply_along_axis(myfunction, axis=1, arr=myarray))
#[ 2352 12672 36992]