如何在 python 中将函数（矩阵 -> 标量）应用于 3 个 numpy 数组

Question

假设我有 3 维 numpy array a，例如如下：import numpy as np a = np.random.randn(3, 3, 3)如何将（矩阵->标量）类型函数应用于a？更具体地说，我想以一种计算效率更高的方式做如下等效的事情：[np.linalg.det(e) for e in a]

Answer 1

np.linalg.det(a)似乎工作得很好，并且运行时间明显更好：

a = np.random.rand(100,3,3)

%timeit -n 100 [np.linalg.det(e) for e in a]

626 µs ± 26.9 µs per loop (mean ± std. dev. of 7 runs, 100 loops each)

%timeit -n 100 np.linalg.det(a)

33.9 µs ± 7.08 µs per loop (mean ± std. dev. of 7 runs, 100 loops each)