python - 如何从二维列表中提取值列表

Question

如何从二维列表中提取满足某些条件的值列表？

如何摆脱空列表和多余的括号？

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

标签： pythonlistarguments

Answer 1

使用列表理解：

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)]