numpy的sum函數(shù)的axis和keepdim參數(shù)詳解

一.axis

axis就是指定軸。

三維數(shù)組可看作元素是二維數(shù)組的一維數(shù)組，二維數(shù)組可看作元素是一維數(shù)組的一維數(shù)組 。（這么理解就舒服了?。?/p>

例：

axis=2就是三維數(shù)組對最里面那一層即每個一維數(shù)組內(nèi)部進(jìn)行求和。

axis=0就是對最外面那一層的元素之間作和。

例子戳這里

二.keepdim

可以理解為'keepdims = True'參數(shù)是為了保持結(jié)果的維度與原始array相同，即keep dimension 保持維度.

import numpy as np
 
b=np.arange(12)
b=b.reshape(2,6)
print(b)
print('b中的元素之和：',np.sum(b))
#即在b的第一個軸上進(jìn)行加和，相當(dāng)于壓縮行，也可以理解為二維矩陣的第一層括號里的東西加和
#若axis=1則是壓縮列，也就是對第二層括號里面的進(jìn)行求和
sum=np.sum(b,axis=0,keepdims=True)
print(sum)

運(yùn)行結(jié)果：

[[ 0  1  2  3  4  5]
 [ 6  7  8  9 10 11]]
b中的元素之和： 66
[[ 6  8 10 12 14 16]]
最后一個輸出指定axis=0,keepdim=True,可看到輸出的是一個二維數(shù)組，如果不加keepdim=True，那么結(jié)果就是一維數(shù)組[ 6  8 10 12 14 16]

到此這篇關(guān)于numpy的sum函數(shù)的axis和keepdim參數(shù)詳解的文章就介紹到這了,更多相關(guān)numpy的sum函數(shù)axis和keepdim參數(shù)內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！