Python itertools库函数大全

1. accumulate

accumulate函数用于返回一个累积的迭代器，它逐个计算传入迭代器中的元素的累积值

itertools中的accumulate函数接受两个参数：

1. iterable：这是要进行累积操作的可迭代对象，例如列表、元组或任何可迭代的数据结构。
2. func (可选)：这是一个二元函数（只接受两个参数），用于指定如何计算累积值。如果未提供此参数，accumulate将默认使用加法来计算累积值。

1.1 对数进行累计计算

import itertools

data = [1, 2, 3, 4, 5]

accumulated_data = itertools.accumulate(data)
print(accumulated_data)

for item in accumulated_data:
    print(item)

import itertools

def custom_multiply(x, y):
    return x * y

data = [1, 2, 3, 4, 5]

accumulated_data = itertools.accumulate(data, custom_multiply)

for item in accumulated_data:
    print(item)

1.2 对字符串进行拼接

import itertools

data = ['a', 'bb', 'ccc', 'dddd']

accumulated_data = itertools.accumulate(data, lambda x, y: x + y)

for item in accumulated_data:
    print(item)

2. chain

将多个可迭代对象连接成一个单一的迭代器。它会逐个迭代每个可迭代对象，并在第一个对象耗尽后继续迭代下一个对象，依此类推，直到所有对象都被迭代完为止。

import itertools

a = [1, 2, 3]
b = ['a', 'b', 'c']

chained_iterable = itertools.chain(a, b)

for item in chained_iterable:
    print(item)

3. combinations（组合）

用于生成指定长度的所有可能的组合，其中组合元素取自于给定的可迭代对象。这个函数返回一个迭代器，该迭代器产生元组，每个元组包含指定长度的元素组合。

import itertools

a = ['a', 'b', 'c', 'd']

# 从a中任选三个元素进行排列
for item in itertools.combinations(a,