如何在Python中测试项目不同部分的内存使用情况？

Question

我尝试在一个脚本中比较不同算法的性能。如果不使用分析器，我们可以使用任何方法来估算每个算法实现结束时的内存使用情况吗？

def main():
    test_case = open('text.txt', 'r')
    for line in test_case.readlines(): #input different datasets
       #do something in Alg1
       #evaluate memory usage in Alg1
       #do something in Alg2
       #evaluate memory usage in Alg2

Answer 1

假设您使用的是Linux或Mac OS X;

您可以使用标准库模块resource将调用进程的总内存消耗量调为bytes：

>>> import resource
>>> resource.getrusage(resource.RUSAGE_SELF).ru_maxrss
19968000