[FIXED] Using np.where for nested lists

Issue

I am trying to use np.where() function with nested lists. I would like to find an index with a given condition of the first layer of the nested list.

For example, if I have the following code

arr = [[1,1], [2,2],[3,3]]
a = np.where(arr == [2,2])

then ideally I would like code to return 'a' as 1. Since [2,2] is in index 1 of the nested list.

However, I am just getting a empty array back as a result.

Of course, I can make it work easily by implementing external for loop such as

for n in range(len(arr)):
   if arr[n] == [2,2]:
      a = n

but I would like to implement this simply within the function np.where(write the entire code here).

Is there a way to do this?

Solution

The best solution is that mentioned by @Michael Szczesny, but using np.where you can do this too:

a = np.where(np.array(arr) == [2, 2])[0]
resulted_ind = np.where(np.bincount(a) == 2)[0]  # --> [1]

Answered By - Ali_Sh