如何解决如何在python中构建对角矩阵?
我有形状为 [100,100] 的矩阵 a 和形状为 [100,100] 的矩阵 b。它们充满了一些值。
我想做的是为a和b的每个元素建立这样的对角矩阵[[a1,0],[0,b1]]。
最好的方法是什么?
我相信预期的形状是数组 c = [2,2,100,100],其中第一个 [2,2] 表示一个对角矩阵的形状,总共有 [100,100] 个这样的数组。
F.e.假设我的 a = [[1,2],[3,4]],b = [[5,6],[7,8]] 。我想得到什么: arr1 = [[1,5]],array2 = [[2,6]],等等..所以,最终的形状是[2,4,4]
谢谢!
解决方法
您写道“对角矩阵矩阵”的最终形状应该是 (2,2,n,n)
但 IMO 您真正想要的是 (n,2)
形状,以便您可以使用 {{1} } 获得
out[i,j]
你可能同意我的不同理解,这里是代码,给你你想要的
[ [a[i,j],0],[0,b[i,j]] ]
应该可以将这个过程推广到任意数量的矩形 N×M 矩阵 In [64]: import numpy as np
...: n = 3 ; nn = n*n
...: a,b = np.arange(2*nn).reshape(2,n)
...: c = np.transpose(np.array((a,b)),(1,0)).reshape(n,2)
...: out = np.zeros((n,2))
...: out[:,:,0] = c[:,0]; out[:,1,1] = c[:,1]
...: print('2,1',a[2,1],b[2,out[2,sep='\n')
...: print('0,0',a[0,b[0,out[0,2',2],sep='\n')
2,1
7
16
[[ 7. 0.]
[ 0. 16.]]
0,0
0
9
[[0. 0.]
[0. 9.]]
0,2
2
11
[[ 2. 0.]
[ 0. 11.]]
In [65]: out.shape
Out[65]: (3,3,2)
,但需要更多的大脑消耗。
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。