解決Numpy中sum函數(shù)求和結(jié)果維度的問題
更新時間：2019年12月06日 10:41:50 作者：紫金港浪里小白條
今天小編大家分享一篇解決Numpy中sum函數(shù)求和結(jié)果維度的問題，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
使用Numpy（下面簡稱np）中的sum函數(shù)對某一維度求和時，由于該維度會在求和后變成一個數(shù)，所以所得結(jié)果的這一維度為空。
比如下面的例子：
a = np.array([[1,2,3],[4,5,6]]) b = np.sum(a,axis=1) print(b.shape) # (2,)
所以，對于一個shape為(2,3)的數(shù)組，在默認(rèn)情況下使用np.sum函數(shù)求和后得到的結(jié)果shape是 (2,)，如果我們想得到的是(2,1)的shape怎么辦？比如Ng的深度學(xué)習(xí)編程練習(xí)中Course 1 Assignment 4就要求這樣。使用reshape函數(shù)當(dāng)然可以，只是沒有必要，太麻煩了一點不優(yōu)雅。我們可以使用通過設(shè)置keepdims參數(shù)實現(xiàn)，還是這個例子：
a = np.array([[1,2,3],[4,5,6]]) b = np.sum(a,axis=1,keepdims=True) print(b.shape) # (2,1)
(2,1)和(2,)的shape之間不同參見 What's the difference between (N,) and (N,1) in Numpy? —Stackoverflow
這里有個小例子可以幫助理解：
a = np.ones((5,)) b = np.ones((5,1)) print(a) # [1. 1. 1. 1. 1.] print(b) # [[1.] # [1.] # [1.] # [1.] # [1.]]
以上這篇解決Numpy中sum函數(shù)求和結(jié)果維度的問題就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關(guān)文章
Centos下實現(xiàn)安裝Python3.6和Python2共存
這篇文章主要介紹了Centos下實現(xiàn)安裝Python3.6和Python2共存,小編覺得挺不錯的，現(xiàn)在分享給大家，也給大家做個參考。一起跟隨小編過來看看吧2018-08-08
Python求算數(shù)平方根和約數(shù)的方法匯總
這篇文章主要介紹了 Python求算數(shù)平方根和約數(shù)的方法匯總的相關(guān)資料,需要的朋友可以參考下2016-03-03
本文主要介紹了python 邊緣擴充方式的實現(xiàn)示例，文中通過示例代碼介紹的非常詳細(xì)，具有一定的參考價值，感興趣的小伙伴們可以參考一下2022-03-03
閾值處理是一種簡單、有效的將圖像劃分為前景和背景的方法。圖像分割通常用于根據(jù)對象的某些屬性(例如，顏色、邊緣或直方圖)從背景中提取對象。本文將為大家詳細(xì)介紹OpenCV中的閾值處理，需要的可以參考一下2022-02-02
這篇文章主要給大家介紹了關(guān)于django主動拋出403異常的相關(guān)資料，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2019-01-01
Python實現(xiàn)PPT/PPTX批量轉(zhuǎn)換成PDF
這篇文章主要為大家詳細(xì)介紹了如何使用Python將PowerPoint演示文稿（PPT、PPTX等）轉(zhuǎn)換為PDF文件,使演示內(nèi)容能夠在更多的設(shè)備上展示,感興趣的小伙伴可以了解下2024-01-01
這篇文章主要介紹了通過幾個機器學(xué)習(xí)中最常用的概率分布為例，來看看如何從一個概率分布中采樣，文章中的代碼對我們的工作或?qū)W習(xí)具有一定價值，感興趣的朋友可以了解一下2021-12-12
python通過pillow識別動態(tài)驗證碼的示例代碼
在上網(wǎng)時，經(jīng)常會遇到驗證碼，本次試驗將帶領(lǐng)大家認(rèn)識驗證碼的一些特性，并利用 Python 中的 pillow 庫完成對驗證碼的破解。感興趣的可以了解一下2021-11-11
這篇文章主要介紹了Python的jsonpath庫使用方法實例,接口返回的jsonn數(shù)據(jù),需要取值后斷言,一般我們是使用jsonpath來提取接口返回的數(shù)據(jù) ,JsonPath是一種信息抽取類庫,是從JSON文檔中抽取指定信息的工具,,需要的朋友可以參考下2023-08-08
本文主要介紹了np.ones的使用小結(jié)，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2023-03-03