您好,登錄后才能下訂單哦!
需求:
原矩陣
[[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的例子就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支持億速云。
免責聲明:本站發布的內容(圖片、視頻和文字)以原創、轉載和分享為主,文章觀點不代表本網站立場,如果涉及侵權請聯系站長郵箱:is@yisu.com進行舉報,并提供相關證據,一經查實,將立刻刪除涉嫌侵權內容。