Question

我有这个函数，我想从大文件中读取一个字节。唯一的问题是，在一定量的文件读取后，PC上的内存会从稳定的1.5gb跳到4gb以上，具体取决于文件读取的数量。（我打破80个文件，因为更高的会让我的电脑崩溃）

我想要的只是得到1个字节，而不是整个文件。请帮助。

    def mem_test():
        count = 0
        for dirpath, dnames, fnames in scandir.walk(restorePaths[0]):
            for f in fnames:
                if 'DIR.EDB' in f or 'PRIV.EDB' in f:
                    f = open(os.path.join(dirpath, f), 'rb')
                    f.read(1) #Problem line!
                    f.close()
                    if count > 80:
                        print 'EXIT'
                        sys.exit(1)                    
                    count += 1
mem_test()

Answer 1

为了解决内存问题，我想你想要使用一个生成器：

def mem_test():
    for dirpath, dnames, fnames in scandir.walk(restorePaths[0]):
        for f in fnames:
            if 'DIR.EDB' in f or 'PRIV.EDB' in f:
                with open(os.path.join(dirpath, f), 'rb') as f:
                    yield f.read(1) #Problem line!

results = [x for x in mem_test()]

从大文件中读取单个字节

1 个答案: