python中numpy矩陣的零填充的示例代碼
需求:
對(duì)于圖像處理中的一些過(guò)程,我需要對(duì)讀取的numpy矩陣進(jìn)行size的擴(kuò)充,比如原本是(4,6)的矩陣,現(xiàn)在需要上下左右各擴(kuò)充3行,且為了不影響數(shù)值計(jì)算,都用0填充。
比如下圖,我有一個(gè)4x5大小的全1矩陣,但是現(xiàn)在我要在四周都加上3行的0來(lái)擴(kuò)充大小,最后擴(kuò)充完還要對(duì)原區(qū)域進(jìn)行操作。



方法:
想到了幾種方法,記錄一下。
一、再new一個(gè)更大的所需要的矩陣大小
a = np.ones((4,5)) #假設(shè)原矩陣是4x5的全1矩陣
print(a.shape)
b_pad = np.zeros((4 + 6, 5 + 6)) #要四周擴(kuò)充3行,所以寬高都要加6
h,w = b.shape
print(b.shape) #(10,11)
b[3:h-3,3:w-3] = a #再把原矩陣放到相應(yīng)位置
print(b)
#如果要對(duì)原數(shù)值一一進(jìn)行操作,可以定位到各元素
for i in range(3,h-3):
for j in range(3,w-3):
b[i][j] = 2
print(b)這個(gè)方法簡(jiǎn)單粗暴,邏輯上比較好理解,但是會(huì)消耗內(nèi)存空間
二、pad函數(shù)
其實(shí)numpy已經(jīng)封裝了一個(gè)函數(shù),就是pad
a = np.ones((4,5))
print(a.shape)
b = np.pad(a,3,'constant') #對(duì)a,上下左右各擴(kuò)充3行,constant缺省,默認(rèn)為0
print(b)
print(b.shape)
#如果只要左上角擴(kuò)充的話
c = np.pad(a,(3,0),'constant')
print(c)
print(c.shape)

其他想法
還有兩種想法,但我沒(méi)實(shí)踐過(guò)。一個(gè)是 二維轉(zhuǎn)成一維,然后用list在每一組的前后可以加0,用append連接,最后再轉(zhuǎn)成二維。這樣子的話,0想放在哪里就可以自己定了。 還有一種是用torch.nn,在pytorch中有對(duì)于卷積的0填充padding。不過(guò)需要把numpy轉(zhuǎn)成tensor張量,最后再轉(zhuǎn)回來(lái),比較麻煩。
到此這篇關(guān)于python中numpy矩陣的零填充的示例代碼的文章就介紹到這了,更多相關(guān)numpy矩陣零填充內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家!
相關(guān)文章
Python中字符串列表的相互轉(zhuǎn)換實(shí)際應(yīng)用場(chǎng)景
在Python編程中,經(jīng)常會(huì)遇到需要將字符串列表相互轉(zhuǎn)換的情況,這涉及到將逗號(hào)分隔的字符串轉(zhuǎn)換為列表,或者將列表中的元素連接成一個(gè)字符串,本文將深入討論這些情景,并提供豐富的示例代碼,幫助讀者更全面地理解字符串列表的轉(zhuǎn)換操作2023-12-12
Python網(wǎng)絡(luò)爬蟲(chóng)之爬取微博熱搜
這篇文章主要介紹了Python網(wǎng)絡(luò)爬蟲(chóng)之爬取微博熱搜的相關(guān)知識(shí),非常不錯(cuò),具有一定的參考借鑒價(jià)值,需要的朋友可以參考下2019-04-04
68行Python代碼實(shí)現(xiàn)帶難度升級(jí)的貪吃蛇
本文主要介紹了Python代碼實(shí)現(xiàn)帶難度升級(jí)的貪吃蛇,文中通過(guò)示例代碼介紹的非常詳細(xì),具有一定的參考價(jià)值,感興趣的小伙伴們可以參考一下2022-01-01
python一鍵升級(jí)所有pip package的方法
下面小編就為大家?guī)?lái)一篇python一鍵升級(jí)所有pip package的方法。小編覺(jué)得挺不錯(cuò)的,現(xiàn)在就分享給大家,也給大家做個(gè)參考。一起跟隨小編過(guò)來(lái)看看吧2017-01-01
Python中函數(shù)eval和ast.literal_eval的區(qū)別詳解
eval函數(shù)在Python中做數(shù)據(jù)類型的轉(zhuǎn)換還是很有用的。它的作用就是把數(shù)據(jù)還原成它本身或者是能夠轉(zhuǎn)化成的數(shù)據(jù)類型。那么eval和ast.literal_val()的區(qū)別是什么呢?本文將大家介紹關(guān)于Python中函數(shù)eval和ast.literal_eval區(qū)別的相關(guān)資料,需要的朋友可以參考下。2017-08-08
python 常見(jiàn)字符串與函數(shù)的用法詳解
這篇文章主要介紹了python 常見(jiàn)字符串與函數(shù)的用法,本文通過(guò)實(shí)例代碼給大家介紹的非常詳細(xì),具有一定的參考借鑒價(jià)值,需要的朋友可以參考下2018-11-11

