두 배열의 거의 동일한 튜플 멤버를 찾고 있습니까?

Question

두 개의 위치 튜플 목록 (x, y 데이터)이 있습니다. 나는 두 배열 또는 두 목록에있는 위치 (또는 튜플)에 대한 인덱스 목록을 반환하고 싶습니다. 그러나 위치 데이터 값은 정확하게 동일하지 않을 것이며, x와 y 좌표 모두에서 +/- 4의 uncertianty가있을 것입니다. 예를 들어

:

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 

어느 하나를 사용하는 것이 잘 될 것입니다.

+/- 값을 지정하여 2 개의 목록에서 일치하는 데이터의 색인을 반환하는 Python 함수가 있습니까?

크기가 같지 않은 ~ 400 개의 튜플로 구성된 3 개의 목록에 대해이 작업을 수행해야합니다.

심지어 같은 것을 사용하는 생각

:

common=[a in A for a in B] 

어떻게 든 a의 범위를 지정, 단지 인덱스를 x와 y 데이터를보고 돌아 대신 참/거짓,하지만 난 정말 돈 이 접근법을 알지 못합니다. 루프는 유일한 방법입니다. 각 값을 따로보고, 차이점을 확인한 후 < 4인지 확인한 다음 색인을 얻으시겠습니까?

Answer 1

무차별 솔루션은 어떻습니까?

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 그러나 효율성이 문제가되지 않는다면 이는 효과가있다.

Answer 2

isclosemath 모듈의 기능을 사용하는 것이 좋습니다.