定义列表时有条件地将项目添加到列表中？

Question

有没有办法在定义列表时有条件地将项目添加到列表中？这就是我的意思：

l = [
    Obj(1),
    Obj(2),
    Separator() if USE_SEPARATORS,
    Obj(3),
    Obj(4),
    Obj(5),
    Separator() if USE_SEPARATORS,
    Obj(6)
]

显然上面的代码不行，请问有没有类似的方法呢？

目前我有

l = [item for item in (
        Obj(1),
        Obj(2),
        Separator() if USE_SEPARATORS,
        Obj(3),
        Obj(4),
        Obj(5),
        Separator() if USE_SEPARATORS,
        Obj(6)
) if not isinstance(item, Separator) or USE_SEPARATORS]

但我想知道是否有其他方法不需要遍历列表，因为它们可能有 10 000 个项目，并且在我执行代码时服务器会停止四分之一秒左右。这是一款第一人称射击游戏，所以四分之一秒实际上可能会对某人的生死产生影响。

Answer 1

我会在之后插入它们；毕竟列表是可变的：

l = [
    HeadObj(1),
    HeadObj(2),
    BodyObj(1),
    BodyObj(2),
    BodyObj(3),
    FooterObj(1)
]
if USE_SEPARATORS:
    l.insert(2, Separator())
    l.insert(6, Separator())

Answer 2

如果不需要，我会添加分隔符并删除它们：

l = [
    Obj(1),
    Obj(2),
    Separator(),
    Obj(3),
    Obj(4),
    Obj(5),
    Separator(),
    Obj(6)]

if not USE_SEPARATORS:
    l = [object for object in l if not isinstance(object, Separator)]

Answer 3

另一种方法是使用 splat/unpacking 运算符将元素展开为分隔符或什么都不展开：

possible_separator = (Separator(),) if USE_SEPARATORS else ()
l = [
    Obj(1),
    Obj(2),
    *possible_separator,
    Obj(3),
    Obj(4),
    Obj(5),
    *possible_separator,
    Obj(6)
]

定义列表时有条件地将项目添加到列表中？

Conditionally add items to a list when defining the list?

python

list-comprehension

list

python-3.x