[파이썬] numpy

numpy를 활용하여 좌표 사이의 거리 구하기

import numpy as np
import math

x1 = 3
y1 = 10

x2 = 5
y2 = 25

a = math.sqrt((x2 - x1) ** 2 + (y2 - y1) ** 2)
print(a)    # 15.132745950421556

def dist(p1, p2):   # python list version
    print('x : ', (p2[0] - p1[0]), (p2[0] - p1[0]) ** 2)
    print('y : ', (p2[1] - p1[1]), (p2[1] - p1[1]) ** 2)
    return math.sqrt((p2[0] - p1[0]) ** 2 + (p2[1] - p1[1]) ** 2)

def dist_np(p1, p2):       # [3, 10]  [5, 25]
    i = p2 - p1            # [2, 5]
    i = sum(i * i)         # [4, 225]
    return math.sqrt(sum((p2 - p1) ** 2))

b = dist([x1, y1], [x2, y2])
print(b)

c = dist_np(np.array([x1, y1]), np.array([x2, y2]))
print(c)