Question

以下是我们今天将使用的两个工具，它们通常很有用。

1） Psutil 是检查内存使用情况的好方法。 这在这里很有用，因为我们使用更大的数据集。

import psutil

process = psutil.Process(os.getpid())
t = process.memory_info()

t.vms, t.rss

# (19475513344, 17856520192)

def mem_usage():
    process = psutil.Process(os.getpid())
    return process.memory_info().rss / psutil.virtual_memory().total

mem_usage()

# 0.13217061955758594

2） TQDM 提供了进度条。

from time import sleep

# 不使用 TQDM
s = 0
for i in range(10):
    s += i
    sleep(0.2)
print(s)

# 45

# 使用 TQDM
from tqdm import tqdm

s = 0
for i in tqdm(range(10)):
    s += i
    sleep(0.2)
print(s)

'''
100%|██████████| 10/10 [00:02<00:00,  4.96it/s]
45
'''