Python 中的平分法或哈希表方法

Question

我想加快我的函数在python中的执行时间。我读到一个很好的方法是使用平分法或哈希表方法。你知道我如何使用这个函数做到这一点吗？

from time import time
import csv
f = open('file.csv')
reader = csv.reader(f, delimiter=';')
def old(abi):
    first = True
    for row in reader:
        if first:
            first = False
            first_row = row
        else:
            if row[0] == abi:
                res = row
                res = dict(zip(first_row, res))
                break
@timing
def test2():
    for x in xrange(3000, 800000):
        old(str(x))
test2()

非常感谢你帮助我;)

Answer 1

我怀疑你的问题是I/O（而不是CPU）绑定的。

如果它确实受 CPU 限制，您可以尝试一件事来提高性能：将 for 循环替换为生成器。这样，迭代将在CPython的C端进行。

def old(path, abi):
    with open(path) as handle:
        r = csv.reader(handle, delimiter=";")
        header = next(r)
        try:
            result = next(row for row in r if row[0] == abi)
            return dict(zip(header, result))
        except StopIterations:
            return None  # Not found.