列表理解和 str.split() 返回列表中的列表而不是单个字符串

Question

假设

thread_sold = ['white', 'white&blue', 'white&blue', 'white', 'white&yellow', 'purple', 'purple&yellow', 'purple&yellow']

我需要该列表中的所有项目，有时用 & 分隔，有时不用。下面的函数有效，但我想知道如何通过列表理解来做到这一点。

def cant_list_comprehend(lst):
    thread_sold_split = []
    for i in lst:
        if "&" in i:
            for x in i.split("&"):
                thread_sold_split.append(x)
        else:
            thread_sold_split.append(i)
    return thread_sold_split

returns ['white', 'white', 'blue', 'white', 'blue', 'white', 'white', 'yellow', 'purple', 'purple', 'yellow', 'purple', 'yellow', 'blue', 'blue', 'purple', 'blue', 'white', 'white'...]

我试过的列表理解：

thread_sold_split_bad = [
    [x for x in y.split("&")] if "&" in y else y for y in thread_sold
]

returns ['white', ['white', 'blue'], ['white', 'blue'], 'white', ['white', 'yellow'], 'purple', ['purple', 'yellow'],...]

如果可以避免的话，我想避免在我的代码中添加命名函数，我也有兴趣使用 lambda 函数来解决这个问题，尽管我目前正在学习基本的东西。

Answer 1

你可以使用functools.reduce:

from functools import reduce

reduce(lambda x, y: x + y, [x.split("&") for x in thread_sold])

或

import itertools

list(itertools.chain.from_iterable(x.split("&") for x in thread_sold))

输出

['white', 'white', 'blue', 'white', 'blue', 'white', 'white', 'yellow', 'purple', 'purple', 'yellow', 'purple', 'yellow']

Answer 2

thread_sold_split_bad = [
    [x for x in y.split("&")] if "&" in y else y for y in thread_sold
]

您上面的尝试不起作用，因为您正在执行嵌套列表推导而不是普通推导（注意主括号内的括号）。这是修正后的版本：

>>> thread_sold_split = [x for y in thread_sold for x in y.split("&")]
>>> print(thread_sold_split)
['white', 'white', 'blue', 'white', 'blue', 'white', 'white', 'yellow', 'purple', 'purple', 'yellow', 'purple', 'yellow']