遍历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().假设您的矩阵是2D，则可以这样使用:

Use numpy.apply_along_axis(). Assuming your matrix is 2D, you can use 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屋！