我希望能够迭代矩阵以将函数应用于每一行。我该如何为 Numpy 矩阵做到这一点？
您可以使用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]
本文内容由网友自发贡献，版权归原作者所有，本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容，请联系:hwhale#tublm.com(使用前将#替换为@)
