給numpy.array增加維度的超簡單方法
更新時間：2021年06月01日 14:44:42 作者：whyume
這篇文章主要介紹了給numpy.array增加維度的超簡單方法，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教
輸入：
import numpy as np a = np.array([1, 2, 3]) print(a)
輸出結(jié)果：
array([1, 2, 3])
輸入：
print(a[None])
輸出結(jié)果：
array([[1, 2, 3]])
輸入：
print(a[:,None])
輸出結(jié)果：
array([[1],
[2],
[3]])
numpy數(shù)組的維度增減方法
使用np.expand_dims()為數(shù)組增加指定的軸,np.squeeze()將數(shù)組中的軸進行壓縮減小維度。
1.增加numpy array的維度
在操作數(shù)組情況下，需要按照某個軸將不同數(shù)組的維度對齊，這時候需要為數(shù)組添加維度(特別是將二維數(shù)組變成高維張量的情況下)。
numpy提供了expand_dims()函數(shù)來為數(shù)組增加維度：
import numpy as np a = np.array([[1,2],[3,4]]) a.shape print(a) >>> """ (2L, 2L) [[1 2] [3 4]] """ # 如果需要在數(shù)組上增加維度,輸入需要增添維度的軸即可，注意index從零還是 a_add_dimension = np.expand_dims(a,axis=0) a_add_dimension.shape >>> (1L, 2L, 2L) a_add_dimension2 = np.expand_dims(a,axis=-1) a_add_dimension2.shape >>> (2L, 2L, 1L) a_add_dimension3 = np.expand_dims(a,axis=1) a_add_dimension3.shape >>> (2L, 1L, 2L)
2.壓縮維度移除軸
在數(shù)組中會存在很多軸只有1維的情況，可以使用squeeze函數(shù)來壓縮冗余維度
b = np.array([[[[5],[6]],[[7],[8]]]])
b.shape
print(b)
>>>
"""
(1L, 2L, 2L, 1L)
array([[[[5],
[6]],
[[7],
[8]]]])
"""
b_squeeze = b.squeeze()
b_squeeze.shape
>>>(2L, 2L) #默認(rèn)壓縮所有為1的維度
b_squeeze0 = b.squeeze(axis=0) #調(diào)用array實例的方法
b_squeeze0.shape
>>>(2L, 2L, 1L)
b_squeeze3 = np.squeeze(b, axis=3) #調(diào)用numpy的方法
b_squeeze3.shape
>>>(1L, 2L, 2L)
以上為個人經(jīng)驗，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關(guān)文章
深入理解python中if?__name__?==?‘__main__‘
很多python的文件中會有語句if?__name=='__main__':,一直不太明白,最近查閱了一下資料,現(xiàn)在明白,本文就來深入理解一下,感興趣的可以了解一下2023-08-08
這篇文章主要介紹了Python教程教你如何去除背景，修飾圖片中的頭發(fā)是設(shè)計師最煩人的任務(wù)之一,下面文章將讓大家學(xué)會怎么輕松摳圖去背景，需要的小伙伴可以參考一下2022-02-02
python中windows鏈接linux執(zhí)行命令并獲取執(zhí)行狀態(tài)的問題小結(jié)
這篇文章主要介紹了python中windows鏈接linux執(zhí)行命令并獲取執(zhí)行狀態(tài),由于工具是pyqt寫的所以牽扯到用python鏈接linux的問題，這里記錄一下一些碰到的問題，需要的朋友可以參考下2022-11-11
這篇文章主要介紹了Python DES加密實現(xiàn)原理及實例解析,文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友可以參考下2020-07-07
今天小編就為大家分享一篇numpy ndarray 取出滿足特定條件的某些行實例，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2019-12-12
Python基于域相關(guān)實現(xiàn)圖像增強的方法教程
當(dāng)在圖像上訓(xùn)練深度神經(jīng)網(wǎng)絡(luò)模型時，通過對由數(shù)據(jù)增強生成的更多圖像進行訓(xùn)練，可以使模型更好地泛化。本文將為大家介紹Python基于域相關(guān)的圖像增強實現(xiàn)方法，需要的可以了解一下2022-01-01
python 環(huán)境搭建 及python-3.4.4的下載和安裝過程
這篇文章主要介紹了python 環(huán)境搭建 python-3.4.4的下載和安裝過程，文中給大家補充介紹了pycharm的基本用法，非常不錯，具有一定的參考借鑒價值 ,需要的朋友可以參考下2019-07-07
這篇文章主要給大家介紹了關(guān)于如何使用PyTorch優(yōu)化一個邊緣檢測器的相關(guān)資料,文中通過代碼介紹的非常詳細(xì),對大家的學(xué)習(xí)或者工作具有一定的參考借鑒價值,需要的朋友可以參考下2024-09-09
python實現(xiàn)dnspod自動更新dns解析的方法
這篇文章主要介紹了python實現(xiàn)的dnspod自動更新dns解析的方法,需要的朋友可以參考下2014-02-02
Python簡單過濾字母和數(shù)字的方法小結(jié)
這篇文章主要介紹了Python簡單過濾字母和數(shù)字的方法,涉及Python基于內(nèi)置函數(shù)與正則表達式進行字母和數(shù)字過濾的相關(guān)操作技巧,需要的朋友可以參考下2019-01-01