Python中numpy數(shù)組的維度增減方法詳解
更新時間：2023年09月23日 09:24:30 作者：頁頁讀
這篇文章主要介紹了Python中numpy數(shù)組的維度增減方法詳解,在操作數(shù)組情況下,需要按照某個軸將不同數(shù)組的維度對齊,這時候需要為數(shù)組添加維度(特別是將二維數(shù)組變成高維張量的情況下),numpy提供了expand_dims()函數(shù)來為數(shù)組增加維度,需要的朋友可以參考下
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) #默認壓縮所有為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)到此這篇關(guān)于Python中numpy數(shù)組的維度增減方法詳解的文章就介紹到這了,更多相關(guān)numpy數(shù)組維度增減內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！
相關(guān)文章
根據(jù)DataFrame某一列的值來選擇具體的某一行方法
今天小編就為大家分享一篇根據(jù)DataFrame某一列的值來選擇具體的某一行方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2018-07-07
在本篇文章里我們給大家整理了關(guān)于學(xué)了python能做哪方面的工作的相關(guān)內(nèi)容，有興趣的朋友們可以學(xué)習(xí)下。2020-06-06
Python運行報錯UnicodeDecodeError的解決方法
本文給大家分享的是在Python項目中經(jīng)常遇到的關(guān)于編碼問題的一個小bug的解決方法以及分析方法，有相同遭遇的小伙伴可以來參考下2016-06-06
python?random模塊常用函數(shù)基礎(chǔ)教程
這篇文章主要為大家介紹了python?random模塊常用函數(shù)基礎(chǔ)教程，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進步，早日升職加薪2023-06-06
從基礎(chǔ)到進階詳解Python處理Word文檔的完全指南
在Python中處理Word文檔是一項常見且實用的任務(wù),本文將介紹如何使用幾個主流的Python庫來創(chuàng)建、修改和處理Word文檔,涵蓋從基礎(chǔ)操作到高級功能的完整流程2026-01-01
Python使用matplotlib繪制三維參數(shù)曲線操作示例
這篇文章主要介紹了Python使用matplotlib繪制三維參數(shù)曲線操作,結(jié)合實例形式分析了Python使用matplotlib的數(shù)值計算與圖形繪制相關(guān)操作技巧,需要的朋友可以參考下2019-09-09
使用python生成隨機驗證碼的方法有很多種，今天小編給大家分享兩種方法，大家可以靈活運用這兩種方法，設(shè)計出適合自己的驗證碼方法2015-12-12
pycharm遠程連接服務(wù)器調(diào)試tensorflow無法加載問題
最近打算在win系統(tǒng)下使用pycharm開發(fā)程序，并遠程連接服務(wù)器調(diào)試程序，其中在import tensorflow時報錯，本文就來介紹一下如何解決，感興趣的可以了解一下2021-06-06
Flaks基礎(chǔ)之在URL中添加變量的實現(xiàn)詳解
這篇文章主要介紹了在python開發(fā)中，F(xiàn)laks框架之上在URL鏈接中添加變量的實現(xiàn)方式，文中通過示例代碼介紹的非常詳細，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2022-09-09
python學(xué)習(xí)之plot函數(shù)的使用教程
這篇文章主要給大家介紹了關(guān)于python學(xué)習(xí)之plot函數(shù)的使用教程，文中通過示例代碼介紹的非常詳細，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2021-04-04