最も近い座標同士を一対一で対応させたいです

前提・実現したいこと

座標が格納されたリストa,bがあり、そこからaとbのお互いの最も近い座標同士を一対一で対応させたいです

発生している問題・エラーメッセージ

実行結果が

[45.967379738244816 array([ 38, 139]) array([ 70, 172])]
[24.166091947189145 array([ 60, 150]) array([ 70, 172])]
[155.32224567009067 array([188,  71]) array([ 70, 172])]
[44.68780594300866 array([ 41, 138]) array([ 70, 172])]

となり、本来は

[24.166091947189145 array([ 60, 150]) array([ 70, 172])]

を採用して、次はこの座標を除いてそれぞれを比較し、どちらかのリストの要素がなくなるまでそれを繰り返したいのですがどうしたら良いでしょうか？

該当のソースコード

python
1import numpy as np
2
3a = np.array([[38,139], [60,150], [188, 71], [41, 138]])
4b = np.array([[23,188], [70, 172], [52, 196]])
5
6for i in a:
7    min = np.linalg.norm(i - b[0])
8    for j in b:
9        c = np.linalg.norm(i - j)
10        if min > c:
11            min = c
12            #index_a = np.where(a[:] == i)
13            #index_b = np.where(b[:] == j)
14            result = np.array([min, i, j])
15        else:
16            pass
17    print(result)

試したこと

補足情報（FW/ツールのバージョンなど）

Python 3.7.4
numpy 1.16.4

行動規範の内容に同意します

回答1件

ベストアンサー

こんな感じでしょうか

Python
1import numpy as np
2
3a = np.array([[38,139], [60,150], [188, 71], [41, 138]])
4b = np.array([[23,188], [70, 172], [52, 196]])
5
6# すべての組み合わせの結果を求める。
7res = np.array([
8    [ np.linalg.norm(y - x) for x in b] for y in a
9    ])
10pair = []
11
12while len(a) > 0 and len(b) > 0:
13    # 2次元配列の中から最も数値の小さい要素の番号を求める
14    y, x = np.unravel_index(np.argmin(res), res.shape)
15    # 組み合わせをpairに追加
16    pair.append([res[y,x], a[y], b[x]])
17
18    # 使用した要素を削除
19    a = np.delete(a, y, 0)
20    b = np.delete(b, x, 0)
21    # 使用した要素によって求められた結果も削除
22    res = np.delete(np.delete(res, y, 0), x, 1)
23    print(pair[-1])
24

output
1[24.166091947189145, array([ 60, 150]), array([ 70, 172])]
2[51.24451190127583, array([ 38, 139]), array([ 23, 188])]
3[59.033888572581766, array([ 41, 138]), array([ 52, 196])]

投稿2019/11/03 15:07