我在 Python 中使用 tqdm 来显示控制台进度条。
我有一个来自另一个库的函数,它偶尔会写入 tqdm 循环内的 stdout 和 stderr。我无法破解该函数的源代码。
同时this doc显示如何重定向 sys.stdout,它不容易推广到 stderr,因为我只能将 stdout 或 stderr 之一传递给 tqdm 的 __init__
中的 file=
参数。请引用this question及其 accepted answer获取说明问题的最少代码。
如何将 stdout 和 stderr 重定向到一起?
最佳答案
Python 在标准库中为您提供了一些帮助程序,请查看 contextlib
:
>>> import io, sys
>>> from contextlib import redirect_stdout, redirect_stderr
>>> from tqdm import tqdm
>>> def foo():
... print('spam to stdout')
... print('spam to stderr', file=sys.stderr)
...
>>> out = io.StringIO()
>>> err = io.StringIO()
>>> with redirect_stdout(out), redirect_stderr(err):
... for x in tqdm(range(3), file=sys.__stdout__):
... print(x) # more spam
... foo()
...
100%|██████████| 3/3 [00:00<00:00, 29330.80it/s]
>>> out.getvalue()
'0\nspam to stdout\n1\nspam to stdout\n2\nspam to stdout\n'
>>> err.getvalue()
'spam to stderr\nspam to stderr\nspam to stderr\n'
关于python - 使用 python tqdm 库重定向 stdout 和 stderr,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/45722952/