距離の近い2点をScipyのKDTreeを使って求める - Qiita

問題. 1辺が100の正方形の中の格子点にある10個の点の中から最も近い2点を見つけよ。 このような問題を解くのにScipyのKDTreeを使うと簡単にかつ高速に求めることが出来たので紹介します。 まず10個の点を乱数で発生させリストにします。 from scipy.spatial import KDTree from random import randint N, XYrange = 10, 10**2 points = [(randint(0,XYrange),randint(0,XYrange)) for i in range(N)] print(points) # [(19, 27), (86, 35), (35, 10), (42, 40), (16, 52), (87, 3), (4, 37), (88, 34), (32, 51), (39, 59)]