python - if else 在带有 for 循环的列表理解中

Question

我已经为扁平化列表编写了自己的简单解决方案:

lists = [0, 10, [20, 30], 40, 50, [60, 70, 80], [90, 100, 110, 120]]
new = []
for item in lists:
    if str(item).isdigit() != True:
        for v in item:
            new.append(v)
    else:
        new.append(item)
print(new)

但我想在以下代码中添加 else/elif 以缩短代码:

new = [v for item in lists if str(item).isdigit() != True for v in item]

我不知道如何解决语法错误。

Answer 1

试试这个:

>>> [v for item in lists for v in (item if isinstance(item, list) else [item])]
[0, 10, 20, 30, 40, 50, 60, 70, 80, 90, 100, 110, 120]

这有点低效，因为它将一个整数包装到一个列表中，然后对其进行迭代，但对于大多数用途来说这无关紧要。尽量不要使用字符串表示来检查数据类型，因为您可能会得到意想不到的结果(例如，尝试使用 float 而不是列表中的整数运行原始代码)。