{site_name}

{site_name}

🌜 搜索

Python itertools 模块是用于高效地构建迭代器的模块

Python 𝄐 0
python中itertools,python__lt__函数,python itertuples,python iterate,python中iterator,python中itertools的用法
Python itertools 模块是用于高效地构建迭代器的模块。这个模块提供了许多有用的函数,可以使用它们来进行迭代器操作,比如过滤、组合、排列等等。下面是一些 itertools 函数的例子:

1. itertools.count(start=0, step=1)
这个函数可以用于生成一个无限递增的整数序列。可以指定起始值和步长。
python
import itertools

for num in itertools.count(1, 2):
if num > 10:
break
print(num)

输出:

1
3
5
7
9


2. itertools.cycle(iterable)
这个函数可以用于创建一个无限循环的迭代器,重复一个可迭代对象中的所有元素。
python
import itertools

colors = ['red', 'green', 'blue']
color_cycle = itertools.cycle(colors)

for i in range(6):
print(next(color_cycle))

输出:

red
green
blue
red
green
blue


3. itertools.compress(data, selectors)
这个函数可以用于从一个序列中选择出满足条件的元素,selectors 序列可以控制哪些元素被选中。
python
import itertools

data = [1, 2, 3, 4, 5]
selectors = [True, False, True, False, True]

result = list(itertools.compress(data, selectors))
print(result)

输出:

[1, 3, 5]


4. itertools.product(*iterables, repeat=1)
这个函数可以用于生成多个可迭代对象的笛卡尔积。
python
import itertools

colors = ['red', 'green', 'blue']
sizes = ['S', 'M', 'L']

for c, s in itertools.product(colors, sizes):
print(c, s)

输出:

red S
red M
red L
green S
green M
green L
blue S
blue M
blue L