python - 使 tqdm 等函数在 Python 中接受不同类型的参数

Question

这不是 tqdm 特有的问题，而是一个有关在 Python 中将参数传递给函数的通用问题。我想实现以下功能，而不必在 tqdm 下复制整个 block 。任何帮助将不胜感激。

if flag == True:
    with tqdm(dataloader, total=args.num_train_batches) as pbar:
else:
    with tqdm(dataloader) as pbar:

更具体地说，我可以通过这样的方式传递参数吗？

if flag == True:
    tqdm_args = dataloader, total=args.num_train_batches
else:
    tqdm_args = dataloader
with tqdm(tqdm_args) as pbar:

Answer 1

这实际上很简单，因为他们在制作Python时似乎就想到了这一点。您可以使用Python的ternary operator为此，请将上面的内容压缩为一行:

with tqdm(dataloader, total=args.num_train_batches if flag else None) as pbar:
  # ...

编辑:用您提到的首选方法回答，是的。这也是有可能的。如果将这些参数放入列表(或字典，如果有关键字参数)，然后在列表的前面放入 * (或 ** 对于字典) name 当调用该函数时，它将列表解压缩为一组参数。

使用列表的示例:

if flag: # if flag is a boolean, putting "== True" does nothing
    tqdm_args = [dataloader, None, args.num_train_batches]
else:
    tqdm_args = [dataloader]
with tqdm(*tqdm_args) as pbar:
    # ...

字典示例:

if flag:
  tqdm_kwargs = {"iterable": dataloader, "total": args.num_train_batches}
else:
  tqdm_kwargs = {"iterable": dataloader}
with tqdm(**tqdm_kwargs) as pbar:
  # ...

很高兴为您提供帮助!