python 双重循环处理

最新推荐文章于 2025-06-10 14:02:41 发布

原创最新推荐文章于 2025-06-10 14:02:41 发布 · 5.2k 阅读

8 ·

CC 4.0 BY-SA版权

文章标签：

#python #开发语言

python开发基础同时被 2 个专栏收录

5 篇文章

订阅专栏

日常开发问题

5 篇文章

订阅专栏

本文探讨了在Python开发中如何优化循环结构，包括使用`itertools`模块简化双重循环，利用推导式展开复杂嵌套数据，以及在已知集合大小时遵循外小内大的循环原则。通过这些方法，可以显著减少代码冗余，提高执行速度。文章还提供了实际代码示例和性能对比。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

序⾔

在开发过程中经常会遇到对可迭代对象进⾏for循环的操作，甚⾄双重编写for循环进⽽找到匹配条件的数据，

某些逻辑上这些操作往往⽆法避免，但可以通过⼀些⽅法使得代码不那么冗余和提升代码效率

循环的情况

1、循环⽐较

双重循环获取对象并进⾏⽐较

>>> a1 = [1, 3, 5, 2, 10]

>>> a2 = [5, 6, 1, 8]

>>> for i in a1:

        for j in a2:
            if i < j:
                print(i,j)

...

(1, 5)

(1, 6)

(1, 8)

(3, 5)

(3, 6)

(3, 8)

(5, 6)

(5, 8)

(2, 5)

(2, 6)

(2, 8)
复制代码

可以通过itertools包对此种循环进⾏优化，不仅使得代码复杂度降低，也提升了执⾏速度

a1 = [1, 3, 5, 2, 10]

a2 = [5, 6, 1, 8]

@count_time
def base_func():

    for i in a1:
        for j in a2:
            if i < j:
                print(i, j)

@count_time
def itertools_func():
    for i in itertools.product(a1, a2):
        if i[0] < i[1]:
            print(i)
            
            
if __name__ == '__main__':
    base_func()
    itertools_func()
    
# 8.106231689453125e-06
# 4.053115844726562e-06
复制代码

2、循环展开

将复杂嵌套数据内容展开，可以使⽤推导式进⾏实现

@count_time

def base_func():

    result = []
    for a in a1:
        for j in a:
            result.append(j)

@count_time
def itertools_func():
    value = [i for value in a1 for i in value]

if __name__ == '__main__':
    print(base_func())
    print(itertools_func())

# 8.106231689453125e-06
# 4.053115844726562e-06
复制代码

3、已知集合⼤⼩

在已知集合⼤致数量进⾏双重for循环处理数据时，应该采⽤外⼩内⼤的原则，把⼤循环写在⾥⾯，⼩循环写在外边

@count_time
def outer_func():

    for i in range(10):
        for j in range(10000): 
            if i == j:
                print(i, j)

@count_time
def inner_func():

    for i in range(10000):
        for j in range(10):
            if i == j:
                print(i, j)

# 0.003239154815673828
# 0.004751205444335937