給numpy.array增加維度的超簡單方法
發(fā)布日期：2022-03-26 19:15 | 文章來源：腳本之家
輸入：
import numpy as np a = np.array([1, 2, 3]) print(a)
輸出結果：
array([1, 2, 3])
輸入：
print(a[None])
輸出結果：
array([[1, 2, 3]])
輸入：
print(a[:,None])
輸出結果：
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)#默認壓縮所有為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)驗，希望能給大家一個參考，也希望大家多多支持本站。
版權聲明：本站文章來源標注為YINGSOO的內(nèi)容版權均為本站所有，歡迎引用、轉載，請保持原文完整并注明來源及原文鏈接。禁止復制或仿造本網(wǎng)站，禁止在非maisonbaluchon.cn所屬的服務器上建立鏡像，否則將依法追究法律責任。本站部分內(nèi)容來源于網(wǎng)友推薦、互聯(lián)網(wǎng)收集整理而來，僅供學習參考，不代表本站立場，如有內(nèi)容涉嫌侵權，請聯(lián)系alex-e#qq.com處理。
相關文章
上一篇：
關注官方微信