为asyncio创建tqdm进度条
为asyncio创建tqdm进度条
提问于 2020-04-05 18:32:13
我正在尝试tqdm进度条，并收集了异步任务。
希望进度条在任务完成时逐步更新。已尝试代码：
import asyncio
import tqdm
import random
async def factorial(name, number):
f = 1
for i in range(2, number+1):
await asyncio.sleep(random.random())
f *= i
print(f"Task {name}: factorial {number} = {f}")
async def tq(flen):
for _ in tqdm.tqdm(range(flen)):
await asyncio.sleep(0.1)
async def main():
# Schedule the three concurrently
flist = [factorial("A", 2),
factorial("B", 3),
factorial("C", 4)]
await asyncio.gather(*flist, tq(len(flist)))
asyncio.run(main())...but这只是完成了tqdm bar，然后处理阶乘。
有没有办法让进度条在每个asyncio任务完成时移动？
回答 4
Stack Overflow用户
回答已采纳
发布于 2020-04-07 14:08:06
对Dragos的pbar格式代码做了几处小改动，并使用tqdm.write()获得了我想要的内容，如下所示：
import asyncio
import random
import tqdm
async def factorial(name, number):
f = 1
for i in range(2, number + 1):
await asyncio.sleep(random.random())
f *= i
return f"Task {name}: factorial {number} = {f}"
async def tq(flen):
for _ in tqdm.tqdm(range(flen)):
await asyncio.sleep(0.1)
async def main():
flist = [factorial("A", 2),
factorial("B", 3),
factorial("C", 4)]
pbar = tqdm.tqdm(total=len(flist), position=0, ncols=90)
for f in asyncio.as_completed(flist):
value = await f
pbar.set_description(desc=value, refresh=True)
tqdm.tqdm.write(value)
pbar.update()
if __name__ == '__main__':
asyncio.run(main())Stack Overflow用户
发布于 2020-09-24 16:15:33
从tqdm version 4.48.0开始，可以使用tqdm.asyncio.tqdm.as_completed()
import tqdm.asyncio
...
for f in tqdm.asyncio.tqdm.as_completed(flist):
await fStack Overflow用户
发布于 2020-04-05 18:43:23
现在，我对asyncho不是特别熟悉，尽管我已经在Python语言中成功地使用过tqdm来处理多进程。对代码的以下更改似乎可以同时更新进度条并打印结果，这可能足以让您开始使用。
responses = [await f
for f in tqdm.tqdm(asyncio.as_completed(flist), total=len(flist))]上面的代码应该会替换main定义中的await asyncio.gather(*flist, tq(len(flist)))。
有关更多信息，以上内容的灵感来自于asyncio aiohttp progress bar with tqdm
为了只打印一次条形图并对其进行更新，我执行了以下操作，这将更新进度条的描述，使其包含您的消息：
import asyncio
import tqdm
async def factorial(name, number):
f = 1
for i in range(2, number + 1):
await asyncio.sleep(1)
f *= i
return f"Task {name}: factorial {number} = {f}"
async def tq(flen):
for _ in tqdm.tqdm(range(flen)):
await asyncio.sleep(0.1)
async def main():
# Schedule the three concurrently
flist = [factorial("A", 2),
factorial("B", 3),
factorial("C", 4)]
pbar = tqdm.tqdm(total=len(flist))
for f in asyncio.as_completed(flist):
value = await f
pbar.set_description(value)
pbar.update()
if __name__ == '__main__':
asyncio.run(main())页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/61041214
复制相关文章
相似问题
腾讯云开发者
