快捷導(dǎo)航

python中numpy矩陣的零填充的示例代碼

更新時(shí)間：2022年08月09日 10:41:49 作者：Eric加油學(xué)！

本文主要介紹了python中numpy矩陣的零填充的示例代碼，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧

需求：

對(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)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: