作者：閑歡

來源：Python 技術(shù)

作為程序員，我們經(jīng)常會(huì)遇到比較耗時(shí)的操作，這個(gè)時(shí)候我們大多數(shù)人會(huì)無助地等待程序執(zhí)行完成，有些人會(huì)趁機(jī)摸一下魚，以便渡過這個(gè)無聊看起來又有點(diǎn)未知的時(shí)間，我就是這樣做的。

但是，我們也可以選擇另一種方式——用一個(gè)炫酷的進(jìn)度條，來觀察處理進(jìn)度，也可以及時(shí)了解程序運(yùn)行的情況，做到心中有數(shù)。

今天就給大家介紹 Python 的一個(gè)庫—— tqdm ，它就能非常完美的支持和解決這些問題，可以實(shí)時(shí)輸出處理進(jìn)度而且占用的CPU資源非常少，支持windows、Linux、mac等系統(tǒng)，支持循環(huán)處理、多進(jìn)程、遞歸處理、還可以結(jié)合linux的命令來查看處理情況，等進(jìn)度展示。

安裝

我們常見的安裝方式是直接 pip 安裝：

pip install tqdm

但是我用這種方式安裝的時(shí)候報(bào)錯(cuò)：

看報(bào)錯(cuò)的意思是這個(gè) pypi.org 不是可信網(wǎng)站，于是我加上了：

pip install tqdm --trusted-host pypi.org

結(jié)果還是這個(gè)報(bào)錯(cuò)。

這有點(diǎn)無語了。我接著嘗試使用豆瓣的源來進(jìn)行安裝：

pip install -i https://pypi.douban.com/simple tqdm

還是一樣的報(bào)錯(cuò)。

但是我瀏覽器訪問這個(gè)網(wǎng)站沒問題：https://pypi.org/project/tqdm/

既然這樣，我就把文件下載下來再安裝吧。

下載下來之后，我將 whl 文件放在我的項(xiàng)目目錄，然后使用命令行安裝：

pip install --trusted-host pypi.org tqdm-4.62.3-py2.py3-none-any.whl

這回安裝沒報(bào)錯(cuò)。我再用命令檢查一下安裝是否成功：

pip show tqdm

命令行成功顯示信息：

Name: tqdm Version: 4.62.3 Summary: Fast, Extensible Progress Meter Home-page: https://tqdm.github.io Author: None Author-email: None License: MPLv2.0, MIT Licences Location: c:pworkspacemypyvenvlibsite-packages Requires: colorama Required-by:

折騰了半天，終于安裝成功了！

使用實(shí)例

本文的示例都是在 jupyter notebook 環(huán)境下運(yùn)行的，不同環(huán)境運(yùn)行的效果會(huì)有差別。

我們進(jìn)入 tqdm 的源碼，可以找到 __init__ 方法：

def __init__(self, iterable=None, desc=None, total=None, leave=True, file=None,
                 ncols=None, mininterval=0.1, maxinterval=10.0, miniters=None,
                 ascii=None, disable=False, unit='it', unit_scale=False,
                 dynamic_ncols=False, smoothing=0.3, bar_format=None, initial=0,
                 position=None, postfix=None, unit_divisor=1000, write_bytes=None,
                 lock_args=None, nrows=None, colour=None, delay=0, gui=False,
                 **kwargs):

從中我們可以看到 tqdm 支持很多參數(shù)，下面列一些常見的參數(shù)：

• iterable：可迭代的對(duì)象，在手動(dòng)更新時(shí)不需要進(jìn)行設(shè)置。
• desc：字符串，左邊進(jìn)度條描述文字。
• total：總的項(xiàng)目數(shù)。
• leave：布爾值，迭代完成后是否保留進(jìn)度條。
• file：輸出指向位置，默認(rèn)是終端, 一般不需要設(shè)置。
• ncols：調(diào)整進(jìn)度條寬度，默認(rèn)是根據(jù)環(huán)境自動(dòng)調(diào)節(jié)長度，如果設(shè)置為0，就沒有進(jìn)度條，只有輸出的信息。
• unit：描述處理項(xiàng)目的文字，默認(rèn)是'it'，例如: 100 it/s，處理照片的話設(shè)置為'img' ，則為 100 img/s。
• unit_scale：自動(dòng)根據(jù)國際標(biāo)準(zhǔn)進(jìn)行項(xiàng)目處理速度單位的換算，例如 100000 it/s >> 100k it/s。

基于可迭代對(duì)象使用

直接將可迭代對(duì)象傳入作為參數(shù)，我們來看一下例子：

from tqdm import tqdm from time import sleep for char in tqdm(['h', 'e', 'l', 'l', 'o']):
    sleep(0.25) for i in tqdm(range(100)):
    sleep(0.05)

實(shí)現(xiàn)的進(jìn)度條效果如下：

這里面的 tqdm(range()) 我們也可以用 tqdm 提供的 trange() 來代替，可以簡化代碼。

from tqdm.notebook import trange for i in trange(100):
    sleep(0.05)

實(shí)現(xiàn)的進(jìn)度條效果如下：

我們還可以為進(jìn)度條添加描述：

pbar = tqdm(range(5)) for char in pbar:
    pbar.set_description("Progress %d" %char) sleep(1)

實(shí)現(xiàn)的進(jìn)度條效果如下：

接下來，我要改變一下進(jìn)度條的顏色：

手動(dòng)控制進(jìn)度條

我們可以使用 with 語句來手動(dòng)控制進(jìn)度條。

with tqdm(total=100) as pbar: for i in range(1, 5): sleep(1) # 更新進(jìn)度 pbar.update(10*i)

這里我設(shè)置進(jìn)度條的更新的間隔，設(shè)置總數(shù)為 total=100，然后分四次，使得進(jìn)度條按 10%，20%，30%，40%的間隔來更新。

實(shí)現(xiàn)的進(jìn)度條效果如下：

接下來， 我要改變一下進(jìn)度條的顏色：

with tqdm(total=100, colour='yellow') as pbar: for i in range(1, 5): sleep(1) # 更新進(jìn)度 pbar.update(10*i)

我把進(jìn)度條改為了黃色。

接下來，我們來個(gè)嵌套進(jìn)度條玩玩。

for i in trange(3, desc='outer loop'): for i in trange(100, desc='inner loop', leave=False): sleep(0.01)

這里我用兩層 for 循環(huán)實(shí)現(xiàn)了嵌套，并且將內(nèi)層的 參數(shù) leave 設(shè)置為 False ，意思是內(nèi)層的進(jìn)度條每執(zhí)行一次都會(huì)消失。

我們來看看最終運(yùn)行的效果：

總結(jié)

這個(gè)進(jìn)度條庫的使用方法是不是很簡單，幾行代碼就能為我們的程序增色不少。心動(dòng)不如行動(dòng)，趕快用起來吧！