python 实现矩阵填充0的例子
需求：
原矩阵
[[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的例子就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持软件开发网。
您可能感兴趣的文章:Python实现螺旋矩阵的填充算法示例Python实现的矩阵类实例Python numpy.zero() 初始化矩阵实例python+numpy实现的基本矩阵操作示例Python表示矩阵的方法分析
