python - NumPy 向量化函数，形状未知

Question

我有一个以 numpy 数组作为参数的函数，例如:

def f(arr):
  return arr.sum()

并且我想从 A 中的每个 vec 创建 numpy 数组，因此如果 A.shape = (14,12,7)，我的函数 myfunc(A).shape = ( 14,12)
IE。myfunc(A)[x, y] = f(A[x, y])

请注意，未指定 len(A.shape)。

Answer 1

您可以申请sum沿着最后一个轴:

A.sum(axis=-1)

例如:

In [1]: np.ones((14,12,7)).sum(axis=-1).shape
Out[1]: (14, 12)

如果您有通用函数，您可以使用 apply_along_axis :

np.apply_along_axis(sum, -1, A)