Python:有没有办法使用 timeit.timeit() 导入变量？

Question

假设我有一些函数接受一个数组并将每个元素更改为 0。

def function(array):
    for i in range(0,len(array)):
        array[i] = 0
return array

我想测试这个函数在随机数组上运行需要多长时间，我希望在 timeit 测试之外生成它。换句话说，我不想把生成数组的时间计入时间。

我首先将一个随机数组存储在变量 x 中并执行:

timeit.timeit("function(x)",setup="from __main__ import function")

但这给了我一个错误:NameError: global name 'x' is not defined

我该怎么做？

Answer 1

从__main__以及导入x:

timeit.timeit("function(x)", setup="from __main__ import function, x")

和function一样，x是__main__模块中的一个名字，可以导入到timeit中> 设置。