时间戳

首页 > 解决方案 > 如何估计迭代时间并用 for 循环的 tqdm 显示？

python - 如何估计迭代时间并用 for 循环的 tqdm 显示？

问题描述

我想估计以下代码的迭代持续时间并将其显示在进度条（tdqm）上。这个函数将给出所有可能的组合string.printable。

import strings
from itertools import combinations

def generate_combinations(iterable):
  lst = list(iterable)
  for lenght in range(0, len(lst) + 1):
    for i in combinations(lst, lenght):
      with open('passwords.txt', 'a') as f:
        w = ''.join(map(str, i))
        f.write(w + '\n')

generate_combinations(string.printable)

我认为你可以使用timeand来做到这一点tqdm。

预期输出：

estimated time: 12 seconds left
tqdm progress bar: 12% |||||||...

标签： pythontimeitertoolstqdm

解决方案

最终答案（@jq170727 的建议）是以下代码：

import string
from tqdm.auto import trange, tqdm
from itertools import permutations

请求的生成器（在 google colab 上，使用 TPU：大约 10k it / s）

def combined(lst, lenStart, lenEnd):
  for i in trange(lenStart, lenEnd + 1, desc='Progress'):
    for subset in tqdm(permutations(lst, i),  desc='Subsets', leave = False):
      w = ''.join(map(str, subset))
      yield w

在这里，我们得到了相同的生成器，效率更高（在 google colab 上，使用 TPU：大约 950k / s）

def compress_permutation(lst, lenStart, lenEnd):
  yield ','.join([''.join(map(str, subset)) for i in trange(lenStart, lenEnd + 1, desc='Progress') for subset in tqdm(permutations(lst, i), desc='Subsets', leave = False)])

iterable = list(string.printable[:94])
lenStart, lendEnd = 1, 9

for subset in compress_permutation(iterable, l1, l2):
  with open('passwords.txt', 'a') as f:
    f.write(subset)

推荐阅读