将多个函数应用于 numpy 数组的每一行

Question

假设我们有一个 3 行 2 列的矩阵 as mat，我想在每 3 行上应用what_functions_to_apply_list我定义的 list 函数之一。所以输出 np.apply_along_axis应该是函数输出维度的 3 行。如果不以矢量化方式进行循环，我该如何做到这一点？例如def f1(inp1,inp2):     return out1, out2在哪里 functions_dic = {'f1': func1, 'f2':func2, 'f3':func3} what_functions_to_apply_list = ['f1','f1','f2'] funcs_inputs = [[inp11,inp12], [inp21,inp32], [inp31,inp32]] mat = np.ones((3, 2)) np.apply_along_axis(what_functions_to_apply_list , 1, mat)

Answer 1

函数列表对数组行的直接应用：

In [418]: alist = [np.add, np.subtract, np.multiply]                            

In [419]: data = np.arange(6).reshape(3,2)                                      

In [420]: [foo(*ab) for foo, ab in zip(alist, data)]                            

Out[420]: [1, -1, 20]