如何找到与给定坐标对应的二维网格的索引？

Question

So, I'm having trouble trying to search for the location of a cordinate on a meshgrid (two 2d arrays). A simple example looks like:

>>> x = [1,2,3]  
    y = [4,5,6]
    xx,yy = np.meshgrid(x,y)

    coord = [1,5] #the coordinate I am looking for

    print(xx)
    print(yy)

[[1 2 3]
 [1 2 3]
 [1 2 3]]

[[4 4 4]
 [5 5 5]
 [6 6 6]]

I would like the code to return coord_idx = [1,0] , the index where the coordinate is stored, such that [xx[coord_idx],yy[coord_idx]] returns [1,5] .

Answer 1

Try this:

import numpy as np

x = [1,2,3]
y = [4,5,6]
xx,yy = np.meshgrid(x,y)

coord = [1,5]

coord_idx = np.argwhere((xx==coord[0]) & (yy==coord[1]))[0]