numpyのsum関数の軸とkeepdimパラメーターの詳細な説明

A.axis

axisは指定された軸です。

三次元配列は、要素が二次元配列である一次元配列と見なすことができ、二次元配列は、要素が一次元配列である一次元配列と見なすことができる。（このように理解するのは快適です！）

例：

axis = 2は、3次元配列が最も内側の層、つまり各1次元配列の内側を合計することを意味します。

axis = 0は、最外層の要素を合計することです。

スタンプの例はこちら

 

Two.keepdim

'keepdims = True'パラメータは、結果の次元を元の配列と同じに保つこと、つまり、keepdimensionが次元を維持することであると理解できます。

import numpy as np


b=np.arange(12)
b=b.reshape(2,6)
print(b)
print('b中的元素之和：',np.sum(b))
#即在b的第一个轴上进行加和，相当于压缩行，也可以理解为二维矩阵的第一层括号里的东西加和
#若axis=1则是压缩列，也就是对第二层括号里面的进行求和
sum=np.sum(b,axis=0,keepdims=True)
print(sum)

演算結果：

[[ 0  1  2  3  4  5]
 [ 6  7  8  9 10 11]]
b中的元素之和： 66
[[ 6  8 10 12 14 16]]

最後の出力はaxis = 0、keepdim = Trueを指定し、出力が2次元配列であることがわかります。keepdim= Trueを追加しない場合、結果は1次元配列になります[6 8 10 12 14 16]