Numpy.prod问题
本文最后更新于:2021年1月8日 晚上
import numpy as np
a=[[ [1,2,3,4],[1,2,3,4] ]]
print(np.array(a).shape) # (1, 2, 4)
s=np.prod(a) # 576
s1=np.prod(a,axis=1) # [[ 1,4,9,16]]
s2=np.prod(a,axis=2) # [[24, 24]]
- np.prod在默认情况下计算所有元素的乘积。
- 多维时,指定了axis,在特定轴内相乘
- axis=1,1×1=1,2×2=4,3×3=9,4×4=16
- axis=2,1×2×3×4=24,1×2×3×4=24
本博客所有文章除特别声明外,均采用 CC BY-SA 4.0 协议 ,转载请注明出处!