python中的itertools、functools、operator函数模块详解

清疚 2023-10-16 09:42 28阅读 0赞

![54922cdaf83d4af58d6330ef417d9778.png][]

### **itertools模块** ###

在接口自动化工作中，我们经常需要处理大量的数据和进行各种迭代操作。为了提高效率和简化代码，Python提供了itertools模块，它包含了一些用于创建高效循环迭代器的函数。接下来，让我们一起了解并使用itertools模块，给出在实际接口自动化工作中的五个示例代码。

**示例1：迭代器连续计数**

在接口测试中，有时我们需要生成一系列连续的数字。itertools.count()函数可以帮助我们创建一个无限的迭代器，每次迭代时生成一个递增的整数。以下是一个示例代码：

import itertools
    for num in itertools.count(start=1, step=1):
        print(num)
        if num == 10:
            break

在上述代码中，我们使用count()函数创建一个从1开始、步长为1的迭代器，通过循环打印出前10个数字。

**示例2：迭代器循环重复**

有时我们需要重复一个序列或者迭代器的内容。itertools.cycle()函数可以帮助我们创建一个无限循环的迭代器，每次迭代时重复序列中的元素。以下是一个示例代码：

import itertools
    seq = [1, 2, 3]
    for num in itertools.cycle(seq):
        print(num)
        if num == 3:
            break

在上述代码中，我们使用cycle()函数创建一个无限循环的迭代器，每次迭代时重复序列seq中的元素，直到迭代到数字3为止。

**示例3：迭代器组合排列**

有时我们需要对序列进行组合或者排列，以生成所有可能的组合或者排列结果。itertools.combinations()和itertools.permutations()函数可以帮助我们实现这个目的。以下是一个示例代码：

import itertools
    seq = [1, 2, 3]
    # 组合
    for combination in itertools.combinations(seq, r=2):
        print(combination)
    # 排列
    for permutation in itertools.permutations(seq, r=2):
        print(permutation)

在上述代码中，我们使用combinations()函数生成序列seq中长度为2的所有组合，并使用permutations()函数生成序列seq中长度为2的所有排列。

**示例4：迭代器条件切片**

有时我们需要根据条件对序列进行切片，只保留符合条件的元素。itertools.takewhile()函数可以帮助我们实现这个目的。以下是一个示例代码：

import itertools
    seq = [1, 2, 3, 4, 5, 6]
    # 保留小于等于3的元素
    result = list(itertools.takewhile(lambda x: x <= 3, seq))
    print(result)

在上述代码中，我们使用takewhile()函数保留序列seq中小于等于3的元素，将结果存储在列表中。

**示例5：迭代器按条件筛选**

有时我们需要根据条件从序列中筛选出符合条件的元素。itertools.filterfalse()函数可以帮助我们实现这个目的。以下是一个示例代码：

import itertools

seq = \[1, 2, 3, 4, 5, 6\]

\# 筛选出非偶数的元素

result = list(itertools.filterfalse(lambda x: x % 2 == 0, seq))

print(result)

在上述代码中，我们使用filterfalse()函数从序列seq中筛选出非偶数的元素，将结果存储在列表中。

### **functools模块** ###

在实际的接口自动化工作中，我们经常需要处理函数和可调用对象，而Python的functools模块提供了一些方便的函数和工具，用于高阶函数和可调用对象的操作。接下来，让我们一起了解并使用functools模块，给出在实际接口自动化工作中的五个示例代码。

**示例1：偏函数**

在接口自动化工作中，有时我们需要创建一个新的函数，该函数是对原始函数的部分参数进行了固定的版本。functools模块的partial()函数可以帮助我们实现这个目的。以下是一个示例代码：

import functools
    # 原始函数
    def add(a, b):
        return a + b
    # 创建一个固定了第一个参数的新函数
    add_5 = functools.partial(add, 5)
    # 调用新函数
    result = add_5(10)
    print(result)

在上述代码中，我们使用partial()函数创建了一个新的函数add\_5，它是对原始函数add()的固定参数版本。我们固定了第一个参数为5，并将其作为新函数的默认参数，然后可以使用新函数进行计算。

**示例2：函数缓存**

在接口自动化工作中，某些函数可能会被频繁调用，而这些函数的计算结果是相同的。为了提高效率，我们可以使用functools模块的lru\_cache()函数对函数进行缓存，以避免重复计算。以下是一个示例代码：

import functools
    # 需要进行缓存的函数
    @functools.lru_cache()
    def get_data(url):
        # 发送HTTP请求并获取数据
        # ...
        return data
    # 第一次调用函数，会执行实际的计算
    result1 = get_data("http://example.com")
    # 第二次调用函数，会直接返回缓存的结果
    result2 = get_data("http://example.com")

在上述代码中，我们使用lru\_cache()函数对get\_data()函数进行了缓存。第一次调用函数时，会执行实际的计算并将结果缓存起来，而后续的调用则直接返回缓存的结果，从而提高了函数的执行效率。

**示例3：函数装饰器**

在接口自动化工作中，我们可能需要对某些函数进行额外的操作或者添加功能。functools模块的wraps()函数可以帮助我们创建函数装饰器，用于在不修改原函数代码的情况下，对函数进行包装。以下是一个示例代码：

import functools
    # 自定义装饰器
    def log_decorator(func):
        @functools.wraps(func)
        def wrapper(*args, **kwargs):
            print("Calling function:", func.__name__)
            result = func(*args, **kwargs)
            print("Function", func.__name__, "finished")
            return result
        return wrapper
    # 应用装饰器
    @log_decorator
    def calculate(a, b):
        return a + b
    # 调用被装饰的函数
    result = calculate(5, 10)
    print(result)

在上述代码中，我们定义了一个装饰器函数log\_decorator，它使用wraps()函数保留了被装饰函数的元数据，然后创建了一个新的函数wrapper来包装原函数。在wrapper函数中，我们可以添加额外的操作，比如打印日志。最后，我们使用@log\_decorator语法将装饰器应用到函数calculate上。

**示例4：函数比较器**

在接口自动化工作中，我们可能需要对一组对象进行排序或者查找最大/最小值。functools模块的cmp\_to\_key()函数可以帮助我们将旧式的比较函数转换为键函数，从而在排序和查找操作中使用。以下是一个示例代码：

import functools
    # 旧式的比较函数
    def compare(a, b):
        if a < b:
            return -1
        elif a > b:
            return 1
        else:
            return 0
    # 将旧式的比较函数转换为键函数
    key_func = functools.cmp_to_key(compare)
    # 使用键函数进行排序
    data = [4, 2, 6, 1, 3]
    data.sort(key=key_func)
    print(data)
    # 使用键函数查找最大/最小值
    max_value = max(data, key=key_func)
    min_value = min(data, key=key_func)
    print("Max:", max_value)
    print("Min:", min_value)

在上述代码中，我们定义了一个旧式的比较函数compare，然后使用cmp\_to\_key()函数将其转换为键函数key\_func。通过传递键函数给sort()、max()和min()函数，我们可以在排序和查找操作中使用旧式的比较函数。

**示例5：函数重试**

在接口自动化工作中，有时我们需要对某些函数进行重试，以处理可能的失败情况。functools模块的retry()函数可以帮助我们实现函数的重试逻辑。以下是一个示例代码：

import functools
    # 需要重试的函数
    @functools.retry()
    def send_request(url):
        # 发送HTTP请求
        # ...
        # 如果请求失败，抛出异常
        raise Exception("Request failed")
    # 调用带重试逻辑的函数
    try:
        send_request("http://example.com")
    except Exception as e:
        print("Error:", str(e))

在上述代码中，我们使用retry()函数对send\_request()函数进行了重试逻辑的包装。如果函数抛出异常，retry()函数会自动进行重试，直到达到指定的重试次数。通过这种方式，我们可以处理函数调用可能的失败情况。

### **operator模块** ###

大家在进行Python编程时，经常会用到各种运算符来进行数值计算、逻辑判断等操作。然而，有时候我们需要将这些运算符以函数的形式进行处理，以便更加灵活地应用在不同的场景中。这时，Python的operator模块就派上了用场。operator模块提供了一系列的函数，用于替代常见的运算符，使我们可以以函数的方式进行运算。本文将介绍operator模块的使用，并提供五个实际接口自动化工作中的示例代码。

**示例1：比较运算符**

在接口自动化工作中，我们经常需要比较两个值的大小或者相等关系。使用operator模块的函数可以方便地进行比较操作。下面是一个示例代码：

import operator
    a = 10
    b = 5
    # 使用函数进行比较
    result1 = operator.lt(a, b)  # a < b
    result2 = operator.eq(a, b)  # a == b
    result3 = operator.gt(a, b)  # a > b
    print(result1)  # 输出False
    print(result2)  # 输出False
    print(result3)  # 输出True

在上述代码中，我们使用lt()、eq()和gt()函数来比较变量a和b的大小关系。通过使用operator模块的函数，我们可以清晰地表达比较操作。

**示例2：算术运算符**

在接口自动化工作中，我们有时需要对数字进行算术运算，例如加法、减法、乘法等。operator模块提供了一系列函数来替代这些算术运算符。以下是一个示例代码：

import operator
    a = 10
    b = 5
    # 使用函数进行算术运算
    result1 = operator.add(a, b)  # a + b
    result2 = operator.sub(a, b)  # a - b
    result3 = operator.mul(a, b)  # a * b
    print(result1)  # 输出15
    print(result2)  # 输出5
    print(result3)  # 输出50

在上述代码中，我们使用add()、sub()和mul()函数来进行加法、减法和乘法运算。通过使用operator模块的函数，我们可以避免直接使用运算符，使代码更加清晰易懂。

**示例3：逻辑运算符**

在接口自动化工作中，我们经常需要进行逻辑运算，例如与、或、非等。operator模块提供了相应的函数来进行逻辑运算。以下是一个示例代码：

import operator
    a = True
    b = False
    # 使用函数进行逻辑运算
    result1 = operator.and_(a, b)  # a and b
    result2 = operator.or_(a, b)  # a or b
    result3 = operator.not_(a)  # not a
    print(result1)  # 输出False
    print(result2)  # 输出True
    print(result3)  # 输出False

在上述代码中，我们使用and\_()、or\_()和not\_()函数来进行与、或和非的逻辑运算。通过使用operator模块的函数，我们可以更加灵活地处理逻辑运算。

**示例4：序列操作符**

在接口自动化工作中，我们可能需要对序列（如列表、元组）进行一些操作，例如拼接、重复等。operator模块提供了相应的函数来进行序列操作。以下是一个示例代码：

**import operator**

**list1 = \[1, 2, 3\]**

**list2 = \[4, 5, 6\]**

**\# 使用函数进行序列操作**

**result1 = operator.concat(list1, list2) \# list1 + list2**

**result2 = operator.mul(list1, 3) \# list1 \* 3**

**print(result1) \# 输出\[1, 2, 3, 4, 5, 6\]**

**print(result2) \# 输出\[1, 2, 3, 1, 2, 3, 1, 2, 3\]**

在上述代码中，我们使用concat()函数进行列表的拼接操作，使用mul()函数进行列表的重复操作。通过使用operator模块的函数，我们可以更加方便地对序列进行操作。

**示例5：索引和切片操作符**

在接口自动化工作中，我们经常需要对列表、字符串等进行索引和切片操作。operator模块提供了相应的函数来进行索引和切片操作。以下是一个示例代码：

import operator
    list1 = [1, 2, 3, 4, 5]
    # 使用函数进行索引和切片操作
    result1 = operator.getitem(list1, 2)  # list1[2]
    result2 = operator.setitem(list1, 1, 10)  # list1[1] = 10
    result3 = operator.delitem(list1, 3)  # del list1[3]
    print(result1)  # 输出3
    print(result2)  # 输出None
    print(result3)  # 输出None
    print(list1)  # 输出[1, 10, 3, 5]

在上述代码中，我们使用getitem()函数获取列表的指定索引元素，使用setitem()函数设置列表的指定索引元素，使用delitem()函数删除列表的指定索引元素。通过使用operator模块的函数，我们可以更加方便地进行索引和切片操作。

**最后：**下方这份完整的软件测试视频教程已经整理上传完成，需要的朋友们可以自行领取**【保证100%免费】**

![8b0171187dce405e878f274a8cdd9ec0.png][]

#### **软件测试面试文档** ####

我们学习必然是为了找到高薪的工作，下面这些面试题是来自阿里、腾讯、字节等一线互联网大厂最新的面试资料，并且有字节大佬给出了权威的解答，刷完这一套面试资料相信大家都能找到满意的工作。

![a3e4226c935c4242804cbdf26b571bc6.png][]

![58df6cd288094ad8874e814f2477620b.gif][]

[54922cdaf83d4af58d6330ef417d9778.png]: https://img-blog.csdnimg.cn/54922cdaf83d4af58d6330ef417d9778.png
[8b0171187dce405e878f274a8cdd9ec0.png]: https://img-blog.csdnimg.cn/8b0171187dce405e878f274a8cdd9ec0.png
[a3e4226c935c4242804cbdf26b571bc6.png]: https://img-blog.csdnimg.cn/a3e4226c935c4242804cbdf26b571bc6.png
[58df6cd288094ad8874e814f2477620b.gif]: https://img-blog.csdnimg.cn/58df6cd288094ad8874e814f2477620b.gif