如何向量化传递一个函数到两个numpy数组: 3D和2D?
我有两个多维numpy
数组:x
是3D的,y
是2D的。
如果我有一个函数foo(a, b)
,它接受两个二维数组作为输入,我如何传递给foo
我的多维数组,并以向量化的方式遍历x
的3维,以获得foo
的结果列表?
我一直在尝试使用np.vectorize
来做这件事,但是它遍历数组的各行并产生错误,所以我被卡住了。
转载请注明出处:http://www.intsu.net/article/20230526/1073523.html