python - 是否可以将 tqdm 用于不是循环的进程？

Question

我正在编写一个程序，其中一个操作需要几秒钟(加载一个大文件)。幸运的是，它总是需要相同的时间。所以，为用户着想，我想做一个进度条。然而，tqdm似乎是为循环而设计的。假设 sleep(10)是需要时间的过程。我该如何解决这个问题？

我想为一个不是循环的漫长过程制作一个进度条。

from time import time, sleep
from tqdm import tqdm

for i in tqdm([1]):
    sleep(10)

这段代码的问题是进度条将保持为零，然后在过程结束时跳转到 100%。我想要一个在 10 秒内持续变化的进度条。

Answer 1

您可以定义块大小。持续 10 秒的条形假定您已经知道加载文件所需的时间...

尝试这个

from tqdm import tqdm_notebook as tqdm

chunk_size = 1000
total_chunks=nrows/chunk_size

chunks = pd.read_sql_query(query, connection, index_col='index_name',chunksize= chunk_size)
raw_train_data=pd.DataFrame()

with tqdm(total=total_chunks) as pbar:

    for chunk in chunks:
        raw_train_data = pd.concat([raw_train_data, chunk])
        pbar.update(1)