python - 如何在列表理解中添加多个变量

Question

我在列表理解中有一个 if/else 子句，想知道如何添加多个变量。

这是我使用 if/else 语句时的代码:

if start_year == end_year:
    years = [start_year]
else:
    years = [start_year, end_year]

这是我希望它在列表理解中的样子，但它是不正确的，因为它由于逗号分隔而在 end_year 上添加。

years = [start_year if start_year == end_year else start_year, end_year]

谢谢!

Answer 1

if/else 可以合并成 1 行:

years = [start_year] if start_year == end_year else [start_year, end_year]

请注意，这不是“列表理解”，因为没有 for 循环。