Question

我有两个位置元组列表（x，y数据）。我想为两个列表中的位置（或元组）返回2个数组或索引列表。但是，位置数据值不会完全相等，在x和y坐标上都会有+/- 4的不确定性。

例如：

A=[(1168.593,9.874), (1799.244,40.201),(780.533,12.636)]
B=[(1170.909,8.194), (793.149,10.885), (1801.493,41.603)]

它应该返回：

c=[(0,0),(1,2)]

或：

d=[0,1] #indexes for A
e=[0,2] #indexes for B

任何一个都可以使用。

您是否可以使用Python中的函数返回2个列表中匹配数据的索引，同时指定+/-值？

我需要为3个每个约400个元组的列表执行此操作，这些列表的大小不相等。

我在考虑使用类似的东西：

common=[a in A for a in B]

并以某种方式指定a的范围，只是查看x和y数据并返回索引而不是true / false，但我真的不知道如何处理它。循环是唯一的方法，通过分别查看每个值，获得它们之间的差异，并查看它是否＆lt; 4，然后获取索引？

Answer 1

也许您可以考虑使用isclose模块中的函数math？

Answer 2

蛮力解决方案怎么样？

In [5]: c = []

In [6]: for i, (x1, y1) in enumerate(A):
   ...:     for j, (x2, y2) in enumerate(B):
   ...:         if (x1 - 4 <= x2 <= x1 + 4) and (y1 - 4 <= y2 <= y1 + 4):
   ...:             c.append((i,j))
   ...:
   ...:

In [7]: c
Out[7]: [(0, 0), (1, 2)]

当然，你可以用你想要的任何东西替换条件。可能，使用math.isclose是一个好主意。以矢量化方式执行此操作可能有更好的numpy方式。但如果效率不是问题，这应该有效。

找到几个相同的两个数组的元组成员？

2 个答案: