假設我有 3 維 numpy array a，例如如下：import numpy as np
a = np.random.randn(3, 3, 3)如何將（矩陣->標量）類型函數應用于a？更具體地說，我想以一種計算效率更高的方式做如下等效的事情：[np.linalg.det(e) for e in a]
1 回答
慕工程0101907
TA貢獻1887條經驗 獲得超5個贊
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)
添加回答
舉報
0/150
提交
取消