(67) numpyで配列要素の積

22,443 views

この記事は最終更新から 2424日 が経過しています。

numpy.prod で配列要素の積を算出できる。
軸指定なしならば全要素を対象、axis=Nで軸指定すれば任意方向の要素の積を算出できる。

>>> import numpy as np
>>> a = np.random.randint(0,10,(3,2))
>>> a
array([[9, 4],
       [7, 1],
       [3, 5]])
>>>
>>> a.prod()
3780
>>>
>>> a.prod(axis=0)
array([189,  20])
>>>
>>> a.prod(axis=1)
array([36,  7, 15])