Python中關(guān)于進度條的6個實用技巧

1 簡介
費老師我在幾年前寫過的一篇文章中,介紹過tqdm這個在當(dāng)下Python圈子中已然非常流行的進度條庫,可以幫助我們?yōu)槿魏尉哂醒h(huán)迭代過程的代碼邏輯添加進度條,從而幫助我們感知代碼運行的過程。
而隨著tqdm這幾年來的發(fā)展迭代,更多更好用的功能加入其中,今天的文章中我就給大家總結(jié)了6條非常值得學(xué)習(xí)的tqdm特性。

2 tqdm中實用的6個特性
2.1 autonotebook自動切換進度條風(fēng)格
用過tqdm的朋友們大都知道它可以在常規(guī)的終端以及jupyter風(fēng)格的各種編輯器中使用,且在后者中會以更美觀的形式進行渲染,而以往我們通常需要在常規(guī)的終端里使用from tqdm import tqdm,在jupyter風(fēng)格的編輯器中使用from tqdm.notebook import tqdm來分別導(dǎo)入。
而tqdm最近幾個版本中引入了實驗性質(zhì)的新特性,使得我們只需要統(tǒng)一通過from tqdm.autonotebook import tqdm導(dǎo)入tqdm,就可以自適應(yīng)檢測不同的運行環(huán)境從而自動控制顯示:

2.2 延遲渲染進度條
有時候我們希望當(dāng)循環(huán)過程很快就執(zhí)行完時,可以不打印進度條,畢竟進度條的主要目的是監(jiān)控長時間運行過程,這時我們就可以給tqdm()添加參數(shù)delay來設(shè)置延時的秒數(shù),當(dāng)循環(huán)過程實際運行時長低于delay則無需打印多余的迭代過程:

2.3 自定義進度條色彩
通過為tqdm()設(shè)置參數(shù)colour,可以傳入多種常見色彩格式值,這在jupyter類編輯器中效果尤為明顯:

2.4 自主控制的進度上限
有些情況下,我們傳入tqdm()的對象在迭代過程中是無法預(yù)先計算得到進度上限輪次的,典型如pandas中數(shù)據(jù)框的itertuples(),這種時候我們就可以利用total參數(shù)自行預(yù)設(shè)上限:

2.5 針對enumerate、zip和map的替代
Python中除了常規(guī)的循環(huán)過程以外,還有幾種內(nèi)置函數(shù)也具有迭代循環(huán)的屬性,而tqdm為了方便我們對這些非典型的循環(huán)過程添加進度條,也單獨開發(fā)了tenumerate、tzip以及tmap這三個API,用于替代enumerate、zip和map:

2.6 設(shè)置進度條“用完即逝”
當(dāng)我們希望為多層循環(huán)過程添加進度條監(jiān)視時,常規(guī)的為每一層都直接使用tqdm(),會導(dǎo)致打印出過多的進度條,反而不利于我們觀察進度過程。
而通過使用tqdm.auto中的trange(),我們可以通過設(shè)置參數(shù)leave=False,來讓我們對應(yīng)的進度條加載到頭就自動消失掉,譬如下面動圖中所展示的例子:





