python 實現矩陣填充0的例子
阿新 • • 發佈:2020-01-09
需求:
原矩陣
[[1 2 3] [4 5 6] [7 8 9]]
在原矩陣元素之間填充元素 0,得到
[[1. 0. 2. 0. 3.] [0. 0. 0. 0. 0.] [4. 0. 5. 0. 6.] [0. 0. 0. 0. 0.] [7. 0. 8. 0. 9.]]
思路:
先求出擴充矩陣的維度,再按照每一行每一列遍歷,根據元素的索引規律依次賦值,最終實現新的擴充矩陣。這個思路實現如下:
import numpy as np def PadMat(Ndim,Mat): Brow = Bcol = 2*Ndim-1 B = np.zeros([Brow,Bcol]) for row in range(Brow): if row%2 == 0: for col in range(Bcol): if col%2 == 0: pos_c = int(col/2) pos_r = int(row/2) # print(row,col) B[row,col] = Mat[pos_r,pos_c] else: B[row,col] = 0 return B # A = np.arange(9) + 1 # A = A.reshape([3,3]) A = np.arange(16) + 1 A = A.reshape([4,4]) # print(A.shape[0]) N = Arow = Acol = A.shape[0] NewMat = PadMat(Ndim=N,Mat=A) print(A) print(NewMat)
總結:
這個思路很直接,但是迴圈套迴圈是一個很笨的辦法,而且遍歷也很慢。不知道網友有什麼好的思路嗎?
以上這篇python 實現矩陣填充0的例子就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支援我們。