Python如何修改代码直接获取列表推导生成的非嵌套一维列表

你当前代码返回嵌套列表的原因是赋值时给目标列表额外套了一层方括号，直接移除外层多余的方括号即可实现需求：

def ee(mid_circ = False):
    initial_layout = [1,0] if mid_circ == True else [1,2,3,0]
    return initial_layout

如果不想修改原有的赋值逻辑，也可以在返回时取嵌套列表的第一个元素：

def ee(mid_circ = False):
    initial_layout = [[1,0] if mid_circ == True else [1,2,3,0]]
    return initial_layout[0]

两种修改方式的执行效果一致：

• 传入mid_circ=True时，返回结果为[1, 0]
• 传入mid_circ=False或者不传参数时，返回结果为[1, 2, 3, 0]

内容的提问来源于stack exchange，提问作者ZR-