Python沿多个数组的轴应用

2024-04-18 23:08:00 发布

您现在位置:Python中文网/ 问答频道 /正文

如果我有一个函数f(x),它以单个1d数组作为参数,并生成1d数组作为输出,那么我可以使用numpy.apply_along_axis将该函数应用于2d数组x的每一行,该数组x的每一行的行都是f的有效参数

现在我想用一个带两个参数的函数来做类似的事情。 E、 我有一个函数f(x,y),它以两个一维数组作为参数,还有两个二维数组x,y,都有n行。我想对每一对行应用f,生成一个数组,它又有n行。

如何有效地实现这一点?

我还对变量感兴趣,其中f需要更多参数或涉及更高维数组:

例如f可以取3个x,y,z形状的数组(2,2);(3,);(5,) 并产生形状(4,4)的结果。

我有X,Y,Z形状(50,100,2,2);(50,100,3);(50,100,5) 想要形状的结果(50,100,4,4)


Tags: 函数numpy参数数组事情感兴趣形状apply