我正在尝试编写一个函数,该函数获取一个 numpy 数组 INPUT,并将其列一个接一个地传递给另一个函数。INPUT 数组是一维或二维(不是更多)第二个函数需要一维数组作为参数。(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])
想法是:在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
)
本文收集自互联网,转载请注明来源。
如有侵权,请联系[email protected] 删除。
我来说两句