python - 将多行循环更改为单行循环

Question

这是一种通过for循环追加到列表的方法:

lst = [] 
for i in range(5):
    lst.append(i)

尽管下面的看起来会越来越好:

lst = [i for i in range(5)]

我试图编写与第二种格式相同的以下代码，但我不断出错。谁能帮忙？

filtered_list = []
for childList in source_list:
    filtered_childList = remove_emptyElements(childList)    
    if filtered_childList:   
         filtered_list.append(filtered_childList)

Answer 1

试试这段代码:

# one liner as you asked:
filtered_list = [remove_emptyElements(l) for l in source_list if remove_emptyElements(l)]

# but I think that this will be better:
filtered_list = (remove_emptyElements(l) for l in source_list)
filtered_list = [l for l in filtered_list if l]

更新:要从评论中解决您的问题，您可以使用此代码段:

sequences_result = []
for sequence in sequences:
    for itemset in sequence:
        itemset_result = []
        for item in itemset.split(","):
            itemset_result.append(item.strip())
        sequences_result.append(itemset_result)
print(sequences_result)