您的位置:

提高Python循环效率的小技巧

一、使用列表推导式

在处理单个序列时,可以使用列表推导式代替for循环,以提高效率。


numbers = [1, 2, 3, 4, 5]
squares = [n**2 for n in numbers]

上面的代码将一个数字列表映射为它们的平方,并将结果存储在另一个列表中。使用列表推导式可将简单的循环转换为单个语句。

二、使用生成器表达式

与列表推导式不同,生成器表达式不会创建完整的列表,而是每次请求一个元素时生成一个元素,这有助于节省内存。


numbers = [1, 2, 3, 4, 5]
squares = (n**2 for n in numbers)

上面的代码和前面的列表推导式代码相同,但是使用圆括号而不是方括号来创建生成器表达式。

三、使用enumerate()函数来获取序列的索引

在某些情况下,需要访问序列中的值和索引。for循环和range()函数可用于完成此操作,但是pyhon内置函数enumerate()更加方便。


numbers = [1, 2, 3, 4, 5]
for i, n in enumerate(numbers):
    print(f"Index {i} has value {n}")

上面的代码将打印列表中每个元素的值和索引。

四、提前终止循环

在某些情况下,可以通过在循环中使用break语句来避免不必要的计算一个循环的所有值。例如,如果我们只是想找到列表中的第一个偶数,可以使用以下代码:


numbers = [1, 3, 5, 4, 7, 2, 6]
for n in numbers:
    if n % 2 == 0:
        print(f"The first even number is {n}")
        break

上面的代码应该在到达第7个元素之前打印数字2。因为我们找到了一个偶数,循环就结束了。

五、使用zip()函数同时迭代多个序列

在python中,通过zip()函数可以同时迭代多个序列。它取每个序列的第一个元素组合成一个元组进行迭代,然后取每个序列的第二个元素组合成一个元组进行迭代,依次类推。


names = ["Alice", "Bob", "Charlie"]
ages = [24, 30, 27]
for name, age in zip(names, ages):
    print(f"{name} is {age} years old.")

上面的代码将打印每个人的姓名和年龄。

六、避免在循环中使用range()函数和len()函数

在循环中使用range()函数和len()函数会导致python重复计算序列的长度。最好使用for循环直接迭代序列。例如:


names = ["Alice", "Bob", "Charlie"]
for i in range(len(names)):
    print(names[i])

上面的代码的输出与以下代码相同:


names = ["Alice", "Bob", "Charlie"]
for name in names:
    print(name)

但是,第二个代码片段不需要在每次迭代时计算序列的长度,因此比第一个代码片段更有效率。