1. 程式人生 > 其它 >python模組tqdm(進度條)的使用

python模組tqdm(進度條)的使用

放幾個樣例: 1.

pbar = tqdm(range(1000))

for idx, num in  enumerate(pbar):
    # print(num)
    pbar.set_description('下載速度 %i' % idx)
    pbar.set_postfix(loss=random(), gen=randint(1,999), str='詳細資訊',
                     lst=[1, 2])
from tqdm import tqdm

# assume
dataset_num = 100000
batch_size = 8
epoch = 1
d_loss = 1.2345
g_loss = 1.2345

pbar = tqdm(range(int(dataset_num / batch_size)))

for i in pbar:
    epoch += 1
    d_loss += 1.2345
    g_loss += 1.2345
    pbar.set_description('Epoch: %i' % epoch)
    pbar.set_postfix(d_loss=format(d_loss,'.3f'), g_loss=format(g_loss,'.3f'))

這樣可以在tqdm讀條的前後顯示需要的資訊,set_description()設定的是字首,set_postfix()設定的是字尾。 如果需要更加細緻全面美觀的功能,可以使用progress模組。使用tqdm主要就是圖個方便省事。

參考連結:

  1. tqdm pbar.set_postfix使用
  2. Python保留指定位數的小數
個性簽名:時間會解決一切