python - numpy 在矩阵数组上应用函数

Question

假设我有一个 2D 矩阵数组（即 3D 数组）

a = np.random.rand(10,2,2)

现在我想在这 10 个2x2矩阵上应用一个函数。例如

def test(i):
    return i.shape
b  = np.array([test(i) for i in a])

因此，b将是 10 个元组的数组(2,2)。现在有没有任何 numpy 方法可以在没有显式循环的情况下做到这一点？我尝试过apply_along_axis，但只在单个轴上应用该功能。

标签： pythonarraysnumpy