Question

我有四个mxn矩阵。从这些矩阵中，我想创建一个矩阵的mxn矩阵，这样新矩阵的每个元素将是一个2x2矩阵，并从每个矩阵中获取各个元素。到目前为止，我可以考虑堆叠所有四个矩阵，然后沿z轴重塑每个元素并将其替换。我相信对此有更好的解决方案。样本输入：

$$
\begin{bmatrix}
A_{11} & A_{12}\\ 
A_{21} & A_{22}
\end{bmatrix}
,

\begin{bmatrix}
B_{11} & B_{12}\\ 
B_{21} & B_{22}
\end{bmatrix}
,

\begin{bmatrix}
C_{11} & C_{12}\\ 
C_{21} & C_{22}
\end{bmatrix}
,

\begin{bmatrix}
D_{11} & D_{12}\\ 
D_{21} & D_{22}
\end{bmatrix}
$$

预期输出：

$$
\begin{bmatrix}
\begin{bmatrix}
A_{11} & B_{11}\\ 
C_{11} & D_{11}
\end{bmatrix} & \begin{bmatrix}
A_{12} & B_{12}\\ 
C_{12} & D_{12}
\end{bmatrix}\\ 
\begin{bmatrix}
A_{21} & B_{21}\\ 
C_{21} & D_{21}
\end{bmatrix} & \begin{bmatrix}
A_{22} & B_{12}\\ 
C_{22} & D_{22}
\end{bmatrix}
\end{bmatrix
$$

Answer 1

示例创建功能：

In [510]: def foo(astr,m,n):
     ...:     alist = [astr+'%d%d'%(i,j) for i in range(m) for j in range(n)]
     ...:     return np.array(alist).reshape(m,n)
In [511]: foo('A',2,2)
Out[511]: 
array([['A00', 'A01'],
       ['A10', 'A11']], dtype='<U3')

4个此类数组的列表：

In [512]: alist = [foo('A',2,2),foo('B',2,2),foo('C',2,2),foo('D',2,2)]

各种堆叠方式：

In [513]: np.stack(alist)
Out[513]: 
array([[['A00', 'A01'],
        ['A10', 'A11']],

       [['B00', 'B01'],
        ['B10', 'B11']],

       [['C00', 'C01'],
        ['C10', 'C11']],

       [['D00', 'D01'],
        ['D10', 'D11']]], dtype='<U3')

In [514]: np.stack(alist,2)
Out[514]: 
array([[['A00', 'B00', 'C00', 'D00'],
        ['A01', 'B01', 'C01', 'D01']],

       [['A10', 'B10', 'C10', 'D10'],
        ['A11', 'B11', 'C11', 'D11']]], dtype='<U3')
In [515]: _.shape
Out[515]: (2, 2, 4)

这可以通过多种方式重塑：

In [516]: __.reshape(2,2,2,2)
Out[516]: 
array([[[['A00', 'B00'],
         ['C00', 'D00']],

        [['A01', 'B01'],
         ['C01', 'D01']]],


       [[['A10', 'B10'],
         ['C10', 'D10']],

        [['A11', 'B11'],
         ['C11', 'D11']]]], dtype='<U3')
In [517]: _.reshape(4,2,2)
Out[517]: 
array([[['A00', 'B00'],
        ['C00', 'D00']],

       [['A01', 'B01'],
        ['C01', 'D01']],

       [['A10', 'B10'],
        ['C10', 'D10']],

       [['A11', 'B11'],
        ['C11', 'D11']]], dtype='<U3')

您可以创建一个轴，然后将其转置到心脏的内容中。

Answer 2

由于所有元素都具有mxn个元素，因此需要进行mxn个迭代，并在每次迭代时都使用4个矩阵中的每个元素构建一个2x2矩阵，然后将列表添加到最终的mxn矩阵中。

matrix1 = [[1, 1, 1, 1], [1, 1, 1, 1], [1, 1, 1, 1]]
matrix2 = [[2, 2, 2, 2], [2, 2, 2, 2], [2, 2, 2, 2]]
matrix3 = [[3, 3, 3, 3], [3, 3, 3, 3], [3, 3, 3, 3]]
matrix4 = [[4, 4, 4, 4], [4, 4, 4, 4], [4, 4, 4, 4]]

matrixFinal = []

for i in range(len(matrix1)):
    matrixFinal.append([])
    for j in range(len(matrix1[i])):
        matrixFinal[i].append ([[matrix1[i][j], matrix2[i][j]], [matrix3[i][j], matrix4[i][j]]])

print(matrixFinal)

创建每个元素本身都是矩阵的Matrix（array）

2 个答案: