python - 如何在python中构建对角矩阵?
问题描述
我有形状为 [100,100] 的矩阵 a 和形状相同的矩阵b [ 100,100]。它们充满了一些值。
我想要做的是为 a 和 b 的每个元素构建这样的对角矩阵 [[a1,0],[0,b1]]。
最好的方法是什么?
我相信预期的形状是数组 c = [2,2,100,100],其中第一个 [2,2] 表示一个对角矩阵的形状,总共有 [100,100] 个这样的数组。
Fe 假设我的 a = [[1,2],[3,4]], b = [[5,6],[7,8]] 。我想要得到:arr1 = [[1,0],[0,5]], array2 = [[2,0],[0,6]] 等等.. 所以,最终的形状是[2,2,4,4]
谢谢!
解决方案
您写道,“对角矩阵矩阵”的最终形状应该是(2, 2, n, n)
,但 IMO 你真正想要的是一个(n, n, 2, 2)
形状,这样你就可以使用它out[i,j]
来获得
[ [a[i,j], 0],
[0, b[i,j]] ]
您可能同意我的不同理解,这里的代码可以满足您的需求
In [64]: import numpy as np
...: n = 3 ; nn = n*n
...: a, b = np.arange(2*nn).reshape(2,n,n)
...: c = np.transpose(np.array((a,b)),(1,2,0)).reshape(n,n,2)
...: out = np.zeros((n,n,2,2))
...: out[:,:,0,0] = c[:,:,0]; out[:,:,1,1] = c[:,:,1]
...: print('2,1',a[2,1],b[2,1],out[2,1],sep='\n')
...: print('0,0',a[0,0],b[0,0],out[0,0],sep='\n')
...: print('0,2',a[0,2],b[0,2],out[0,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, 2)
应该可以将这个过程推广到任意数量的 N×M 矩形矩阵a, b, c, ..., n, ...
,并消耗更多的大脑。
推荐阅读
- python - 如何在sqlalchemy中的值更改后触发事件
- postgresql - 无法连接到 postgres docker 容器(本地主机)
- javascript - 如何将消息转换为 utf-8
- python - ImportError:无法从部分初始化的模块“电报”导入名称“机器人”
- flutter - 如何使用 Flutter 单击按钮转到新页面
- javascript - React 渲染时间太长——我应该选择哪种架构?
- c++ - 在 NMRU 的实现中我做错了什么?
- wpf - WPF - Open Message Box while another Message Box is open
- c# - 如果多列结果,如何使用 SqlCommand.ExecuteReader 重构包含循环?
- asp.net - 如何从外部世界访问我在 Windows Azure VM 上的 ASP.NET Web API?