np.where（）を使用して複雑な2D Python配列の要素のインデックスを見つけることができません

Question

numpy.where（）を使用して要素のインデックスを調べようとしていますが、空の配列を返します。

import numpy as np 
grid= np.mgrid[-2:2:5*1j, -2:2:11*1j] 
X , Y = grid[0], grid[1] 
complex_grid = X+1j*Y 
xid, yid= np.where(complex_grid == -1.0 - 0.8j) 
print(xid, yid) 

インデックス（1,3）は返されますが、空の配列とそのデータ型が返されます。 私は何が間違っていますか？

EDIT： - 私の主な目的は、グリッドから与えられた座標（x、y）に対応するインデックスを見つけることです。複雑なグリッドを作ったのは、グリッドから得られる2つの2Dマトリックスを融合できるからです。

Answer 1

浮動小数点値はまれではありませんので、一般的に浮動小数点値を直接比較することは悪い考えです。

あなたは、たとえば、値に与えられた近さをチェックするためにabs(difference) < epsilonのようなものを使用することができますしかし：

>>> xid, yid= np.where(np.abs(complex_grid -(-1.0 - 0.8j)) < 1e-10) 
>>> print(xid, yid) 
[1] [3] 

またはより良い：使用numpy.iscloseはすでにそれを行うが、相対および絶対許容誤差とnan取り扱いを可能にします同様に（あなたはこれらを必要とする場合）：対等に山車を比較すると、常にdiceyあるので

>>> xid, yid= np.where(np.isclose(complex_grid, -1 - 0.8j))) 
>>> print(xid, yid) 
[1] [3] 

Answer 2

isclose()は、より良い選択です。

xid, yid = np.where(np.isclose(complex_grid, np.ones(complex_grid.shape) * (-1.0 - 0.8j)))