Python 行内 for 循环 (list comprehension)

引言

有很多读者在学习 Python 会遇到行内 for 循环的写法，这种写法相对于传统的 for 循环会比较难理解，但是同时可以减少代码的行数。行内 for 循环的英文名称为 list comprehension。本文我们将对这一形式进行说明。

正文

列表类型

列表中的元素是有有一定规律的数字

比如，我们想要创建一个列表，其存放有 0，1，2 ... 10 的数字。我们可以采用下面的代码获取这个列表。

x = [i for i in range(11)]
print(x)  # [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10]

列表中的元素是有有一定规律的字符串

比如，我们想要生成几个变量名称，分别叫做 a_0，a_1，a_2，a_3。

str1 = [f'a_{
     i}' for i in range(4