python 如何使用 numpy aply_along_axis 将参数传递给函数

Question

我想对矩阵中的每一列应用函数。我想使用带参数的函数，但我不知道该怎么做，我尝试的事情以错误结束。

我正在运行的代码

import numpy as np

M = np.array([[1,2,3,4],
              [1,2,3,4],
              [1,2,3,4],
              [1,2,3,4]])

def my_function(arr, arg="default"):
    print arg
    return arr

def my_function_allong_axis(M, argument):
    return np.apply_along_axis(my_function, axis=0, arr=M, arg=argument)

my_function_allong_axis(M, "something else")

这会产生 TypeError: apply_along_axis() got an unexpected keyword argument 'arg'

Answer 1

解决方法:

def my_function_allong_axis(M, argument):
    return np.apply_along_axis(my_function, 0, M, argument)

关键字参数是问题所在，因为旧的 numpy