You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
35 lines
827 B
Python
35 lines
827 B
Python
5 years ago
|
"""
|
||
|
|
Thin wrappers around `itertools`.
|
||
|
|
"""
|
||
|
|
from __future__ import absolute_import
|
||
|
|
from tqdm.auto import tqdm as tqdm_auto
|
||
|
|
from copy import deepcopy
|
||
|
|
import itertools
|
||
|
|
__author__ = {"github.com/": ["casperdcl"]}
|
||
|
|
__all__ = ['product']
|
||
|
|
|
||
|
|
|
||
|
|
def product(*iterables, **tqdm_kwargs):
|
||
|
|
"""
|
||
|
|
Equivalent of `itertools.product`.
|
||
|
|
|
||
|
|
Parameters
|
||
|
|
----------
|
||
|
|
tqdm_class : [default: tqdm.auto.tqdm].
|
||
|
|
"""
|
||
|
|
kwargs = deepcopy(tqdm_kwargs)
|
||
|
|
tqdm_class = kwargs.pop("tqdm_class", tqdm_auto)
|
||
|
|
try:
|
||
|
|
lens = list(map(len, iterables))
|
||
|
|
except TypeError:
|
||
|
|
total = None
|
||
|
|
else:
|
||
|
|
total = 1
|
||
|
|
for i in lens:
|
||
|
|
total *= i
|
||
|
|
kwargs.setdefault("total", total)
|
||
|
|
with tqdm_class(**kwargs) as t:
|
||
|
|
for i in itertools.product(*iterables):
|
||
|
|
yield i
|
||
|
|
t.update()
|