python - 将嵌套 for 循环转换为列表理解

Question

我编写了这样的循环代码:

all = []
for sen_list in sen_lists:
    te = []
    for ele in sen_list:
        boolean = ele == "."
        te.append(boolean)
    all.append(te)

sen_lists = [['a', 'b', 'c', 'd', '.', '.'], ['e', 'f', 'g', 'h']]

上面的代码运行良好...但我想转换列表理解代码。

我这样尝试:

[ele == "." for sen_list in rawdf.TEXT[:10] for ele in sen_list]

但是这段代码是行不通的。

请告诉我。谢谢。

Answer 1

考虑一个嵌套列表理解。或者，换句话说，列表推导 of 个列表推导。

内部的表示您的内部 for 循环。外层代表您的外层 for 循环。

sen_lists = [['a', 'b', 'c', 'd', '.', '.'], ['e', 'f', 'g', 'h']]

res = [[ele == "." for ele in sen_list] for sen_list in sen_lists]

# [[False, False, False, False, True, True], [False, False, False, False]]