Как найти список строк и столбцов индекса в Python - PullRequest
1 голос
/ 09 апреля 2019

Я пытаюсь напечатать индекс строки и столбца в списке.

Я использовал (цикл ниже) это, но это один или другой ..

a = [[0,0,1],
    [0,1,0],
    [1,0,0]]
def a():
x = 0
for sol in solutions:
    print(sol)
    for row in sol:
        print(row)

Я пытаюсьраспечатать
(0,2) (1,1) (2,0)
Индекс 1с
Спасибо

Ответы [ 2 ]

0 голосов
/ 09 апреля 2019

Если вам нравится numpy, вы можете превратить его в массив numpy и использовать argwhere ()

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]]
0 голосов
/ 09 апреля 2019

Вы можете использовать enumerate для генерации индексов для списка:

for row, sublist in enumerate(a):
    for column, item in enumerate(sublist):
        if item:
            print((row, column))

Это выводит:

(0, 2)
(1, 1)
(2, 0)
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...