найти положение элемента в матрице

Question

У меня есть список списков.Я хочу найти положение строки "1" в одном конкретном столбце col:

from numpy import transpose
col = 0
lists = [["0", "0", "N"], ["1", "0", "N"], ["N", "N", "N"]]
pos = transpose(lists)[col].index("1")

Но я получаю ошибку:

AttributeError: 'numpy.ndarray' object has no attribute 'index'

Каким-то образом импорт транспонирования из numpy мешает мнеиспользуйте index ().Как мы можем сделать это легко?

Answer 1

Использование np.transpose преобразует ваш собственный list в np.array, поэтому вы получаете ошибку.

Использование numpy:

np.where(np.array(lists)[:,0]=="1")

Использование собственных списков:

list(zip(*lists))[0].index("1")

list(zip(*lists)) - это родной способ переноса ...