python 实现矩阵填充0的例子
更新时间:2019年11月29日 08:38:50 作者:icaoys
今天小编就为大家分享一篇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爬虫之抓取百度贴吧并存储到本地txt文件改进版
前面已经发了一篇关于百度贴吧抓取的代码,今天我们来看下代码的改进版,参考了上篇抓取糗事百科的思路,给需要的小伙伴们参考下吧2014-11-11Python爬虫实现selenium处理iframe作用域问题
这篇文章主要介绍了Python爬虫实现selenium处理iframe作用域问题,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧2021-01-01
最新评论