python - 匹配两个不同的数组并在python中创建一个新数组

Question

我有两个二维数组，我必须通过第一个列索引匹配的第二个数组创建一个新数组过滤。数组的大小不同。

基本思路如下:

file A

#x   y
1   2
3   4
2   2
5   4
6   4
7   4


file B

#x1    y1
0       1
1       1
11      1
5       1
7       1
My expected output 2D array should look like

#newx     newy
1         1
5         1
7         1

我试过以下方法:

match =[]
for i in range(len(x)):
    if x[i] == x1[i]:
        new_array = x1[i]
        match.append(new_array)

print match

这似乎行不通。请建议一种创建新二维数组的方法

Answer 1

尝试 np.isin .

arr1 = np.array([[1,3,2,5,6,7], [2,4,2,4,4,4]])
arr2 = np.array([[0,1,11,5,7], [1,1,1,1,1]])
arr2[:,np.isin(arr2[0], arr1[0])]
array([[1, 5, 7],
       [1, 1, 1]])

np.isin(arr2[0], arr1[0]) 检查 arr2[0] 的每个元素是否在 arr1[0]。然后，我们使用结果作为 bool 索引数组来选择 arr2 中的元素。