如何用Python列表推导式实现字符串列表的条件拆分与扩展？

需求：将输入列表 values = ['v1,v2', 'v3'] 转换为扁平的输出列表 ['v1', 'v2', 'v3']。

已实现的非列表推导式可行代码：

values = ['v1,v2', 'v3']
parsed_values = []

for v in values:
    if ',' in v:
        parsed_values.extend(v.split(','))
    else:
        parsed_values.append(v)

print(parsed_values)    # ['v1', 'v2', 'v3']

尝试的列表推导式未得到正确结果：

parsed_values = [_ for _ in [v.split(',') if ',' in v else v for v in values]]
# 输出: [['v1', 'v2'], 'v3']

原因是内层推导式生成的是包含子列表和字符串的混合结构，外层循环直接将子列表作为单个元素保留，没有展开。

正确的列表推导式写法 #

通过嵌套循环的列表推导式可实现需求，确保所有元素被扁平处理：

values = ['v1,v2', 'v3']
parsed_values = [item for v in values for item in (v.split(',') if ',' in v else [v])]
print(parsed_values)    # ['v1', 'v2', 'v3']

原理说明 #

• 针对每个元素 v：
  • 若包含逗号，用split(',')生成子列表；
  • 若不包含逗号，将v包装为单元素列表[v]；
• 外层循环遍历这个临时列表的每个元素，最终把所有元素收集到同一个扁平列表中。

内容的提问来源于stack exchange，提问作者Harshal Parekh