파이썬에서 유클리드 거리로 복소수를 정렬하는 효율적인 방법

Question

나는 유클리드 거리로 정렬해야하는 몇 개의 복소수를 가지고 있습니다. 이 문제는 다음과 같이 해결합니다.

  # A1x = Lowest Point (LP) 
      # B1x = Point 1 (P1) 
      # B4x = Point 2 (P2) 

      C1 = euclidean(A1x, B1x) # Build the distance between LP and P1 
      C4 = euclidean(A1x, B4x) # Build the distance between LP and P2 

      array = np.array([C1, C4]) # Put the distances into an array... 
      array.sort() # ...and sort it. 

      # If the the distance between LP and P1 is the first element in the array 
      # then take P1 as y_max value etc. 

      if C1 == array[0]: 
       y_max = B1x 

       if C4 == array[1]: 
        y_min = B4x 

      if C4 == array[0]: 
       y_max = B4x 

       if C1 == array[1]: 
        y_min = B1x 

이 방법은 3 ~ 4 포인트에서 잘 작동합니다. 그러나, 지금 나는 8-9 점을 얻었고 모든 단일 점에 대해 if 조건을 써야하기 때문에 위에 언급 한 것과 같은 방식으로 조금 더 심해집니다. 그러므로, 여러분이 유클리드 거리로 복소수를 분류하는 더 나은 방법을 안다면 물어볼 것입니다.

Answer 1

, 이것은 간단하다

Ap = np.array([1, 2]) # "lowest point" 
B = np.array([[0,0], [1,1], [2,2], [3,3], [4,4]]) # sample array of points 
dist = np.linalg.norm(B - Ap, ord=2, axis=1) # calculate Euclidean distance (2-norm of difference vectors) 
sorted_B = B[np.argsort(dist)] 

sorted_B는리스트 B의 점을 포함하는 끝이지만 포인트 AP와 유클리드 거리 순서로 정렬한다. 위의 입력의 경우, 출력을 NumPy와 기능이 방법을 사용하면보다 빠르고 효율적 해당하는 파이썬 기능 list.sort를 사용하는 것보다해야한다는

array([[1, 1], 
     [2, 2], 
     [0, 0], 
     [3, 3], 
     [4, 4]]) 

주를 얻을 것입니다.

Answer 2

from functools import partial 

complex_number_list.sort(key=partial(euclidean, A1x)) 

euclidean 대신 abs을 사용할 수도 있습니다. argsort 사용