迭代 Numpy 矩阵行以分别应用一个函数? [英] Iterating over Numpy matrix rows to apply a function each?

问题描述

我希望能够遍历矩阵以将函数应用于每一行.我该如何为 Numpy 矩阵做这件事?

I want to be able to iterate over the matrix to apply a function to each row. How can I do it for a Numpy matrix ?

推荐答案

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

You can use numpy.apply_along_axis(). Assuming that your array is 2D, you can use it like:

import numpy as np
mymatrix = np.matrix([[11,12,13],
                      [21,22,23],
                      [31,32,33]])
def myfunction(x):
    return sum(x)

print(np.apply_along_axis(myfunction, axis=1, arr=mymatrix))
#[36 66 96]

这篇关于迭代 Numpy 矩阵行以分别应用一个函数?的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！