如何解决在numpy中创建特殊结构的nxn矩阵:单位矩阵和同一矩阵的右移1列的总和
给出一个整数n
,我想使n x n
的numpy矩阵具有以下形式。
对于n =4
:
array([[1.,1.,0.,0.],[0.,1.],[1.,1.]])
对于n= 5
:
array([[1.,1.]])
在数学上,它添加了一个单位矩阵,该矩阵本质上是向右移动1列的单位矩阵。这是分解的示例。
如何在输入为numpy
的{{1}}中做到这一点
解决方法
使用numpy
roll
n = 3
a1 = np.diag(np.full(n,1))
out = a1+np.roll(a1,shift=1,axis=1)
out
Out[21]:
array([[1,1,0],[0,1],[1,1]])
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。