如何在Python中从两个ndimensional数组中获得匹配的行？

Question

我有两个numPy数组，它们的行数和列数不同。每个阵列的结构-年、月、日、时间、卫星数、数据值。每个数组都有不同类型的数据。你知道吗

如何比较这两个数组以仅获得两个数组中的公共行，其中比较参数是前5列，然后是具有相应值的两列。例如：

A=[('2015', '1', '1', 0.0, 'G06', 46.29)
 ('2015', '1', '1', 0.0, 'G12', 444.344)
 ('2015', '1', '1', 0.0, 'G14', -99.269)
 ('2015', '1', '1', 0.0, 'G20', 6.874)
 ('2015', '1', '1', 0.0, 'G24', 158.488)
 ('2015', '1', '1', 0.0, 'G25', -60.831)
 ('2015', '1', '1', 0.0, 'G31', -48.234)
 ('2015', '1', '1', 0.0, 'R07', -6.243)]

B=[('2015', '1', '1', 0.0, 'G06', '0.000')
 ('2015', '1', '1', 0.0, 'G12', '0.000')
 ('2015', '1', '1', 0.0, 'G14', '0.000')
 ('2015', '1', '1', 0.0, 'G24', '0.000')
 ('2015', '1', '1', 0.0, 'G25', '0.000')
 ('2015', '1', '1', 0.0, 'G29', '0.000')
 ('2015', '1', '1', 0.0, 'G31', '0.000')]

结果，我想得到：

C=[('2015', '1', '1', 0.0, 'G06', 46.29, '0.000')
 ('2015', '1', '1', 0.0, 'G12', 444.344, '0.000')
 ('2015', '1', '1', 0.0, 'G14', -99.269, '0.000')
 ('2015', '1', '1', 0.0, 'G24', 158.488, '0.000')
 ('2015', '1', '1', 0.0, 'G25', -60.831, '0.000')
 ('2015', '1', '1', 0.0, 'G31', -48.234, '0.000')]

我可以使用循环来完成，但这不是一个有效的解决方案，当数组的行数超过50000行时。你知道吗

Answer 1

使用熊猫：

import pandas as pd
a = pd.DataFrame(A)
b = pd.DataFrame(B)
c = pd.merge(a, b, 'inner', left_on=[0,1,2,3,4], right_on=[0,1,2,3,4])

其中'inner'表示仅当两个数组中都存在键值时才合并。left_on=[0,1,2,3,4]表示将这些列用作键值。你知道吗

结果如下：

In: print(c)
Out: 
      0  1  2  3    4      5_x    5_y
0  2015  1  1  0  G06   46.290  0.000
1  2015  1  1  0  G12  444.344  0.000
2  2015  1  1  0  G14  -99.269  0.000
3  2015  1  1  0  G24  158.488  0.000
4  2015  1  1  0  G25  -60.831  0.000
5  2015  1  1  0  G31  -48.234  0.000