使用2个Python数组

Question

我有两个二维数组，它们都是相同的大小，每个数组中都有不同的值。我希望能够在其中一个数组中找到一个值，比如说所有的2，以及所有2的数值，然后填充另一个数组中相同位置的任何值。你将如何在Python中做到这一点？

Answer 1

如果您正在使用numpy，你可以试试这个：

import numpy as np 
A = np.array([[11,12,13],[21,22,23],[31,32,33]]) 
B = np.array([[ 1, 2, 3],[ 2, 1, 3],[ 1, 3, 2]]) 
B[B==2] = A[B==2] 

这里，B==2创建一个具有True每个小区的布尔矩阵，其中存在一个B2。现在，B[B==2]从B获取所有这些单元格，并将它们设置为A和A[B==2]中的任意一个。

之后，在B每个2被替换从A相应的值：

[[ 1 12 3] 
[21 1 3] 
[ 1 3 33]]