numpyのwhere()にて、unexpectedと出てしまいうまくいかない

Question

numpyのwhereが使いたい

関数として、受け取った座標データのリストの中から特定のデータを含んでいる番地を受け取りたいと思いました。
そこでindexを使おうとしたところエラーが出たので調べてみた結果、numpy.arrayではwhere()を使うということだったのでそちらに書き換えました。

しかしいくつか形式を変更してもうまくいかず質問するに至りました。

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

Traceback (most recent call last):
 line 82, in expand_circle
    print(np.where(pt = pair[0]))
  File "<__array_function__ internals>", line 4, in where
TypeError: where() got an unexpected keyword argument 'pt'

該当のソースコード

def expand_circle(data, pt, min):
    min += 0.05
    pt = np.array(pt)
    
    #半径を増やした状態での座標データのペアを確認
    for pair in itertools.combinations(pt, 2):
        ex_distance = np.linalg.norm(pair[0]-pair[1])
        
        #接触しているペアを抽出
        if(ex_distance <= min * 2):
            (x_dis, y_dis) = (pair[0] - pair[1]) / 2
            print(pair[0])
            print(pair[1])
            print(pt)
            print(np.where(pt = pair[0]))
            index0 = np.where(pt = pair[0])[0][0]
            index1 = np.where(pt = pair[1])[0][0]
            print("距離：" + str(ex_distance - min))
            print("接触：" + str(pair[0]) + "," + str(pair[1]))
            print("座標距離：" + str(pair[0] - pair[1]))
            
            if(x_dis < 0):
                pt[index0][0] = pt[index0][0] - x_dis
                pt[index1][0] = pt[index1][0] + x_dis
            else:
                pt[index0][0] = pt[index0][0] + x_dis
                pt[index1][0] = pt[index1][0] - x_dis
            
            if(y_dis < 0):
                pt[index0][1] = pt[index0][1] - y_dis
                pt[index1][1] = pt[index1][1] + y_dis
            else:
                pt[index0][1] = pt[index0][1] + y_dis
                pt[index1][1] = pt[index1][1] - y_dis
    
    #半径を返却
    return data, pt, min

試したこと

エラーの下になっている「pt」はclassを見てみたところ'numpy.ndarray'だったので、他のファイルで条件を一致させたうえで簡易的な実装をしてみたところうまくいったのでなぜうまくいかないのかわかりません。

以下、試したコードになります。

a = np.array([[1,2],[3,5],[4,1]])
ai = [[1,2],[3,5]]
ai = np.array(ai)
print(type(a))
print(ai)

ab = np.where(a == ai[1])[0][0]
print(ab)

補足情報

拙い部分が多々あると思われますが、これらの情報の他に必要な情報などがあれば追加いたします。
よろしくお願いします。

追記：出力結果です

点の最短距離：1.118033988749895
点と線の最短距離：0.642529405317179
比較最短距離：0.642529405317179
[29 29]
[27 30]
[[43 25]
 [27 51]
 [50 75]
 [38 32]
 [40 54]
 [41 46]
 [15 37]
 [36 14]
 [25 62]
 [12 12]
 [26 45]
 [19 26]
 [45 43]
 [46 52]
 [48 46]
 [29 22]
 [27 41]
 [29 29]
 [47 71]
 [27 30]]

Comments

[melian]

print(np.where(pt = pair[0])) は print(np.where(pt == pair[0])) ではないでしょうか。

Answer 1

            print(np.where(pt = pair[0]))

を

            print(np.where(pt == pair[0]))

に修正しましょう。

Comments

[narimi]

できました、、、ありがとうございます！！

使い方をきちんと理解できておりませんでした。
本当にありがとうございました。

Answer 2

where関数にはptという名前のキーワード引数はありません、という内容のエラーです。
処理内容は確認していませんが、単にnp.where(pt == pair[0])でよいと思います。

Comments

[narimi]

出力結果を追記しましたので見ていただきたいのですが、リストptに格納された座標データ群に対してpair[0]が何番目にあるのか、というのが求めたいことになります。

np.where(pair[0])にするとエラーは出ないのですが、結果がすべて[0,1]になってしまいうまくいきません。。

[8524ba23]

なるほど。回答を修正しました。

[narimi]

その方法でエラーなく進めることができました。。

勉強不足でした。ありがとうございます。