Python - 将矩阵a提取到矩阵b

Question

我想将矩阵A（4,4）提取到矩阵b（7,4）中。

下面矩阵：

>>> Matrix_A = numpy.array([[[10,1.5,-3.8,8.0],[20,10.2,5.2,6.7],[30,0.5,-6.2,-7.1],[40,-0.7,-0.6,-0.5]]]) 
>>> Matrix_A 
array([[[ 10. , 1.5, -3.8, 8. ], 
     [ 20. , 10.2, 5.2, 6.7], 
     [ 30. , 0.5, -6.2, -7.1], 
     [ 40. , -0.7, -0.6, -0.5]]]) 

>>> Matrix_B = numpy.array([[[10,2.7,4.8,-5.8],[15,-1.4,-6.4,8.1],[20,12.4,-7.1,4.9],[25,-1.5,6.2,-4.8],[30,-6.8,0.47,3.8],[35,5.4,-4.8,10.5],[40,16.2,5.7,-8.3]]]) 
>>> Matrix_B 
array([[[ 10. , 2.7 , 4.8 , -5.8 ], 
     [ 15. , -1.4 , -6.4 , 8.1 ], 
     [ 20. , 12.4 , -7.1 , 4.9 ], 
     [ 25. , -1.5 , 6.2 , -4.8 ], 
     [ 30. , -6.8 , 0.47, 3.8 ], 
     [ 35. , 5.4 , -4.8 , 10.5 ], 
     [ 40. , 16.2 , 5.7 , -8.3 ]]]) 

我想结果是：

>>> Matrix_A_extract_from_Matrix_B 
array([[[ 10. , 2.7 , 4.8 , -5.8 ], 
     [ 20. , 12.4 , -7.1 , 4.9 ], 
     [ 30. , -6.8 , 0.47, 3.8 ], 
     [ 40. , 16.2 , 5.7 , -8.3 ]]]) 

而且如果可能的话，我想获得一个包含你可以看到下面的其余矩阵R：

>>> Matrix_R 
array([[[ 15. , -1.4, -6.4, 8.1], 
     [ 25. , -1.5, 6.2, -4.8], 
     [ 35. , 5.4, -4.8, 10.5]]]) 

主要目标是比较Matrix_A和Matrix_A_extract_from_Matrix_B。

我真正的问题是将矩阵与数千行Matrix_A和Matrix B进行比较，它仅仅是一个简化问题的例子。

非常感谢您的帮助。

Answer 1

首先，因为你的数组是三维数组，你应该挤压大小以将它们转换为二维数组以便于计算。

In [27]: Matrix_A = np.squeeze(Matrix_A) 

In [28]: Matrix_B = np.squeeze(Matrix_B) 

然后你可以使用np.in1d找到共同的第一列的索引，并用一个简单的索引中提取它们：

In [29]: Matrix_B[np.in1d(Matrix_B[:, 0],Matrix_A[:, 0])] 
Out[29]: 
array([[ 10. , 2.7 , 4.8 , -5.8 ], 
     [ 20. , 12.4 , -7.1 , 4.9 ], 
     [ 30. , -6.8 , 0.47, 3.8 ], 
     [ 40. , 16.2 , 5.7 , -8.3 ]]) 

In [30]: Matrix_B[np.logical_not(np.in1d(Matrix_B[:, 0],Matrix_A[:, 0]))] 
Out[30]: 
array([[ 15. , -1.4, -6.4, 8.1], 
     [ 25. , -1.5, 6.2, -4.8], 
     [ 35. , 5.4, -4.8, 10.5]])