自動切換 tqdm 進度條的顯示模式

使用 tqdm?還是 tqdm.notebook?不如讓它自動偵測吧

前言

使用 Python 時,若須顯示進度條,常會使用 tqdm 套件來顯示進度條。

介紹

tqdm 是個方便用於顯示進度條的套件,只要簡單的使用 tqdm() 將要迭代的變數包起來,即可直接顯示進度條

1
2
3
from tqdm import tqdm
for i in tqdm(range(10000)):
    ...

然而若使用 Jupyter Notebook 來執行,顯示上會怪怪的,
一般會替換使用專為 Notebook 設計的 tqdm.notebook

1
2
3
from tqdm.notebook import tqdm
for i in tqdm(range(10000)):
    ...

那如果同一份程式碼,要用於指令界面與 Notebook 呢?
正好套件也有提供能自動偵測環境的 tqdm
只要改用 tqdm.auto 就可以了

1
2
3
from tqdm.auto import tqdm
for i in tqdm(range(10000)):
    ...

結語

tqdm.auto 就對了

參考資料

Licensed under CC BY-NC-SA 4.0
comments powered by Disqus
使用 Hugo 建立
主題 StackJimmy 設計