我怎樣才能得到所有元素的產品在一個維數組numpy的

Question

我有一個一個維數組NumPy的：

a = numpy.array([2,3,3]) 

我想有所有元素，18在這種情況下的產物。我能找到這樣做會

的唯一方法：它看起來很漂亮，但不是非常快

b = reduce(lambda x,y: x*y, a) 

（我需要做這個有很多）。

是否有一個這樣做的numpy方法？如果不是，這樣做的最有效方法是什麼？我的真實世界數組有39個浮動元素。

Answer 1

在NumPy的你可以嘗試：

numpy.prod(a) 

對於較大的陣列numpy.arange(1,40)/10.：

array([ 0.1, 0.2, 0.3, 0.4, 0.5, 0.6, 0.7, 0.8, 0.9, 1. , 1.1, 
     1.2, 1.3, 1.4, 1.5, 1.6, 1.7, 1.8, 1.9, 2. , 2.1, 2.2, 
     2.3, 2.4, 2.5, 2.6, 2.7, 2.8, 2.9, 3. , 3.1, 3.2, 3.3, 
     3.4, 3.5, 3.6, 3.7, 3.8, 3.9]) 

您reduce(lambda x,y: x*y, a)需要24.2μs，

numpy.prod(a)需要3.9μs。

編輯：a.prod()需要2.67μs。感謝J.F. Sebastian！

Answer 2

或者，如果數值精度的損失是沒有問題的，我們能做到

>>> numpy.exp(numpy.sum(numpy.log(a))) 
17.999999999999996 

>>> numpy.prod(a) 
18