Как извлечь список значений из 2-го списка - PullRequest
0 голосов
/ 27 декабря 2018

Как извлечь список значений, удовлетворяющих некоторому условию, из списка 2d?

Как избавиться от пустых списков и лишних скобок?

my_list = [[(i,j) for i in list_ if i == 1] for j,list_ in enumerate(grid)]

1 Ответ

0 голосов
/ 27 декабря 2018

Использование списка понимания:

grid = [[10,13,17],[3,5,1],[13,1,12]]
results = [(i, b) for b in range(len(grid)) for i, a in enumerate(grid[b]) if a == 1]

Вывод:

[(2, 1), (1, 2)]
...