嵌套列表上的理解列表(列表列表)

Question

允许在“列表列表”上使用理解列表吗？我想从嵌套列表中提取一个列表。我确实试过这个：

def main():
    a = ['1','2','3']
    b = ['4','5','6']
    c = ['7','8','9']
    board = [a,b,c]
    y = [x for x in board[1][i] if i in range(0,3)]
    print y

但我得到"NameError:名称'i‘未定义“。我使用了错误的语法，或者嵌套列表不能像这样使用？

非常感谢!

Answer 1

列表理解中的嵌套循环与嵌套常规for循环的工作方式相同，一个在另一个中：

y = [x for i in range(3) for x in board[1][i]]

但在本例中，只选择board[1][:]会更简单，并给出相同的结果；中间行的一个副本。

如果需要将表达式应用于该行中的每一列，则只需直接在board[1]上循环：

y = [foobar(c) for c in board[1]]