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 协议 ,转载请注明出处!