Вы можете использовать enumerate
для создания индексов для списка:
for row, sublist in enumerate(a):
for column, item in enumerate(sublist):
if item:
print((row, column))
Это выводит:
(0, 2)
(1, 1)
(2, 0)
Если вам нравится numpy, вы можете превратить его в массив numpy и использовать аргументгде()
import numpy as np
a = [[0,0,1],
[0,1,0],
[1,0,0]]
a = np.array(a)
answer = np.argwhere(a==1)
Это выведет:
[[0 2]
[1 1]
[2 0]]