python - numpy 测量时间 - 语法错误

Question

我想通过 numpy 来测量时间，但我不确定我的参数是否正确。

import numpy as np
import timeit

def svdsolve(a,b):
    u,s,v = np.linalg.svd(a)
    c = np.dot(u.T,b)
    w = np.linalg.solve(np.diag(s),c)
    x = np.dot(v.T,w)
    return x


A_=np.fromfile('dataA.bin',count=-1,dtype=np.float32)
B_=np.fromfile('dataB.bin',count=-1,dtype=np.float32)


s='svdsolve({0},{1})'.format(A,B)

mytime= timeit.Timer(stmt=s,setup='import numpy').timeit(100)

print mytime

现在它给了我:

File "/usr/lib64/python2.7/timeit.py", line 136, in init code = compile(src, dummy_src_name, "exec") File "", line 6 svdsolve([[ 1.86248358e+09 1.54404045e+09] ^ SyntaxError: invalid syntax

此外，我不明白(我也没有找到引用)在计时器上传递哪些参数。 我不知道如何使用重复 timeit(100)

Answer 1

NumPy 数组的 str 和 repr 都不能保证产生可用于重建原始数组的输出。 (format 使用 str，但 repr 没有帮助。)相反，将数组导入到定时代码的命名空间。假设此代码作为脚本运行，那就是

mytime = timeit.Timer(stmt='svdsolve(A, B)',
                      setup='from __main__ import A, B, svdsolve'
                      ).timeit(100)