一.axis
axis就是指定軸。
三維數(shù)組可看作元素是二維數(shù)組的一維數(shù)組,二維數(shù)組可看作元素是一維數(shù)組的一維數(shù)組 。(這么理解就舒服了!)
例:
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)
運行結(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)文章希望大家以后多多支持腳本之家!
您可能感興趣的文章:- 詳解Numpy擴(kuò)充矩陣維度(np.expand_dims, np.newaxis)和刪除維度(np.squeeze)的方法
- numpy np.newaxis 的實用分享
- numpy:np.newaxis 實現(xiàn)將行向量轉(zhuǎn)換成列向量
- np.newaxis 實現(xiàn)為 numpy.ndarray(多維數(shù)組)增加一個軸
- NumPy中的維度Axis詳解
- 淺談numpy 函數(shù)里面的axis參數(shù)的含義