一个循环来迭代一个 numpy 数组列（维度为 1D 或 2D）

Question

我正在尝试编写一个函数，该函数获取一个 numpy 数组 INPUT，并将其列一一传递给另一个函数。INPUT 数组是 1D 或 2D（不是更多）第二个函数需要 1D 数组作为参数。(len(param.shape)==1)我读过一个类似的线程，其中 OP 想要对所有列求和并检查其他条件......这可能需要另一个答案。所需的伪代码操作：def func(INPUT,a,b,...)    for column in INPUT: #whether be a 1D or 2D        result = another_func(column,...)试过这个：问题是如何不检查 func 内 INPUT 数组的维度：if(len(INPUT.shape)==1):    another_func(INPUT,....)elif(len(INPUT.shape)==2):    for c in range(INPUT.shape[1]):        another_func(INPUT[:,c])

Answer 1

这个想法是：在 1d 输入的情况下，转换为 1 列 2d 数组，然后作为 2d 输入。

def func(INPUT, a, b):

    return np.apply_along_axis(

        lambda col: another_func(col, a, b),  # function to apply

        1,  # axis along which to apply; 1 = columns

        np.reshape(np.atleast_2d(H2), (len(H), -1))  # transform 1D->2D, if necessary

    )