我在 Python 中使用 tqdm 来在我们的脚本中显示控制台进度条。 但是，我还必须调用将 print 消息发送到控制台的函数，而且我无法更改这些函数。 通常，在控制台中显示进度条的同时写入控制台会使显示困惑，如下所示:

from time import sleep
from tqdm import tqdm

def blabla():
  print "Foo blabla"

for k in tqdm(range(3)):
  blabla()
  sleep(.5)

这将创建输出:

0%|                                           | 0/3 [00:00<?, ?it/s]Foo
blabla
33%|###########6                       | 1/3 [00:00<00:01,  2.00it/s]Foo
blabla
67%|#######################3           | 2/3 [00:01<00:00,  2.00it/s]Foo
blabla
100%|###################################| 3/3 [00:01<00:00,  2.00it/s]

According to the documentation of tqdm tqdm.write() 方法提供了一种在不破坏显示的进度条的情况下将消息写入控制台的方法。 因此，此代码段提供了正确的输出:

from time import sleep
from tqdm import tqdm

def blabla():
  tqdm.write("Foo blabla")

for k in tqdm(range(3)):
  blabla()
  sleep(.5)

看起来像这样:

Foo blabla
Foo blabla
Foo blabla
100%|###################################| 3/3 [00:01<00:00,  1.99it/s]

另一方面，有这个solution which permits to silence those functions通过非常优雅地将 sys.stdout 重定向到空白中。 这对于使函数静音非常有效。

由于我仍然想在不破坏进度条的情况下显示来自这些函数的消息，因此我尝试通过将 sys.stdout 重定向到 tqdm.write()<>，然后让 tqdm.write() 写入 old sys.stdout。 这导致片段:

from time import sleep

import contextlib
import sys

from tqdm import tqdm

class DummyFile(object):
  file = None
  def __init__(self, file):
    self.file = file

  def write(self, x):
    tqdm.write(x, file=self.file)

@contextlib.contextmanager
def nostdout():
    save_stdout = sys.stdout
    sys.stdout = DummyFile(save_stdout)
    yield
    sys.stdout = save_stdout

def blabla():
  print "Foo blabla"

for k in tqdm(range(3)):
  with nostdout():
    blabla()
    sleep(.5)

但是，这实际上会像以前一样产生更困惑的输出:

0%|                                           | 0/3 [00:00<?, ?it/s]Foo
blabla


33%|###########6                       | 1/3 [00:00<00:01,  2.00it/s]Foo
blabla


67%|#######################3           | 2/3 [00:01<00:00,  2.00it/s]Foo
blabla


100%|###################################| 3/3 [00:01<00:00,  2.00it/s]

仅供引用:在 DummyFile.write() 中调用 tqdm.write(..., end="") 会产生与第一个输出相同的结果搞砸了。

我不明白为什么这行不通，因为 tqdm.write() 应该在写入消息之前管理清除进度条，然后重写进度条。

我错过了什么？

最佳答案

重定向 sys.stdout 总是很棘手，当两个应用程序同时处理它时，它会变成一场噩梦。

这里的技巧是 tqdm 默认打印到 sys.stderr，而不是 sys.stdout。通常，tqdm 有针对这两个特殊 channel 的反混淆策略，但是由于您正在重定向 sys.stdout，tqdm 会混淆，因为文件句柄更改。

因此，您只需要明确指定 file=sys.stdout 给 tqdm 就可以了:

from time import sleep

import contextlib
import sys

from tqdm import tqdm

class DummyFile(object):
  file = None
  def __init__(self, file):
    self.file = file

  def write(self, x):
    # Avoid print() second call (useless \n)
    if len(x.rstrip()) > 0:
        tqdm.write(x, file=self.file)

@contextlib.contextmanager
def nostdout():
    save_stdout = sys.stdout
    sys.stdout = DummyFile(sys.stdout)
    yield
    sys.stdout = save_stdout

def blabla():
  print("Foo blabla")

# tqdm call to sys.stdout must be done BEFORE stdout redirection
# and you need to specify sys.stdout, not sys.stderr (default)
for _ in tqdm(range(3), file=sys.stdout):
    with nostdout():
        blabla()
        sleep(.5)

print('Done!')

我还添加了一些技巧来使输出更好(例如，在不使用 end=''< 的情况下使用="">print() 时没有无用的 \n/)。

/EDIT:事实上你似乎可以在启动tqdm后做stdout重定向，你只需要指定dynamic_ncols=True tqdm.

关于python - 通过 tqdm.write() 重定向 python 脚本中的打印命令，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/36986929/