Python使用NumPy实现矩阵元素间插入0的方法

Evan
Evan
发布时间:2026-10-13 | 2人浏览过
原创

本文介绍在Python中利用Num库将矩阵元素之间填充0的方法。通过计算新矩阵维度并遍历赋值,实现从原矩阵到扩展矩阵的转换。虽然双重循环逻辑直观,但效率较低,适合理解矩阵索引规律及基础实现。

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 相关产品和应用

全部下载

专题列表