代码中有多个使用tqdm和tqdm.tqdm的场景,notebook中如何区分?
时间: 2023-07-24 09:13:40 浏览: 192
tqdm-4.35.0.tar.gz
在 notebook 中区分代码中使用的不同场景,可以通过判断当前的环境来进行区分。`tqdm`库提供了一个方法`notebook`,用于检测当前代码是否在 notebook 环境中运行。
你可以使用以下方式来区分:
```python
from tqdm import tqdm
from tqdm.notebook import tqdm as tqdm_notebook
import sys
if 'ipykernel' in sys.modules:
# 在 notebook 环境中
progress_bar = tqdm_notebook
else:
# 在其他环境中(如命令行)
progress_bar = tqdm
# 使用 progress_bar 进行进度显示
for i in progress_bar(range(100)):
# 代码逻辑
```
在 notebook 环境中,`tqdm.notebook.tqdm`将被导入为 `tqdm_notebook`,然后通过检测是否存在 `ipykernel` 模块来判断是否在 notebook 环境中运行。如果在 notebook 环境中,则使用 `tqdm_notebook` 来显示进度条,否则使用普通的 `tqdm`。
这样就可以根据不同的环境来选择适当的 `tqdm` 函数,实现在不同场景下的进度显示。
阅读全文