numpy的sum函數的axis和keepdim參數詳解
一.axis
axis就是指定軸。
三維數組可看作元素是二維數組的一維數組,二維數組可看作元素是一維數組的一維數組 。(這麼理解就舒服瞭!)
例:
axis=2就是三維數組對最裡面那一層即每個一維數組內部進行求和。
axis=0就是對最外面那一層的元素之間作和。
例子戳這裡
二.keepdim
可以理解為’keepdims = True’參數是為瞭保持結果的維度與原始array相同,即keep dimension 保持維度.
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,可看到輸出的是一個二維數組,如果不加keepdim=True,那麼結果就是一維數組[ 6 8 10 12 14 16]
到此這篇關於numpy的sum函數的axis和keepdim參數詳解的文章就介紹到這瞭,更多相關numpy的sum函數axis和keepdim參數內容請搜索WalkonNet以前的文章或繼續瀏覽下面的相關文章希望大傢以後多多支持WalkonNet!
推薦閱讀:
- Python中np.linalg.norm()用法實例總結
- python數據分析Numpy庫的常用操作
- numpy.sum()的使用詳解
- numpy中np.nanmax和np.max的區別及坑
- Python NumPy教程之數組的創建詳解