数组索引的匹配 Python

Question

我有一个 n 行的多维数组 q 和另一个 n 行的多维数组 q2。这些数组包含相同的行，但顺序不同。如何获得 2 个数组的匹配列表/数组？示例如下：

q=array([a,b,c,d],  #row 1
        [e,f,g,h],  #row 2
        [i,l,m,n])  #row 3

q2=array([e,f,g,h], #row 2
         [a,b,c,d], #row 1
         [i,l,m,n]) #row 3

q_index=[1,2,3]
q2_index=? #I know is [2,1,3] but I can I get it?

我需要创建索引的 q 列表/数组和相关的索引 q2 列表/数组列表。 谢谢

Answer 1

先将q转换为字典，先将item转换为元组作为键使用，索引作为值使用。

>>> q = [['a','b','c','d'], ['e','f','g','h'], ['i','l','m','n']]
>>> q2 = [['e','f','g','h'], ['a','b','c','d'], ['i','l','m','n']]
>>> q_dict = {tuple(x):i for i, x in enumerate(q, 1)}
>>> [q_dict[tuple(item)] for item in q2]
[2, 1, 3]

Answer 2

您可以使用此代码，但请注意它具有二次运行时。鉴于您当前的数据结构，多项式是我们将得到的最好的。对于小的ns，首先将其转换为不同的数据结构可能效率不高。这将取决于n 的大小是否更快或使用哈希表。鉴于数组中的值是浮动的，您必须特别小心：

def compFloatLists(l1, l2, epsilon=0.0001):
    if len(l1) != len(l2):
        return False
    for e1, e2 in zip(l1, l2):
        if abs(e1-e2) > epsilon:
            return False
    return True

q2_index = []
for e1 in q2:
    for i, e2 in enumerate(q1):
        if compFloatLists(e1, e2):
            q2_index.append(q_index[i])
            break

作为不相关的附注：您可能希望将 q_index 重命名为 q_indices，这使您的代码更容易理解恕我直言。