计时Python。它是如何运作的？

Question

我想对一个函数进行计时，并且想使用 timeit 库。我在网上找不到任何好的例子。我必须对 maxcut 库中的函数“largest_eigenvector”进行计时，该函数将图 G 作为输入，该图 G 由 networkx 库中的函数返回。

所以我想对这段代码进行计时：

import maxcut as mc 
import networkx as nx 
G = nx.complete_graph(3)

mc.largest_eigenvector(G)

它显然工作得很好。相比之下，我这样做了：

s = """
    import maxcut as mc 
    import networkx as nx 
    G = nx.complete_graph(3)
    """
t = timeit.Timer(s, 'mc.largest_eigenvector(G)')

但它说： UnboundLocalError：在赋值之前引用局部变量“mc”

我不知道为什么。请有人帮忙，这只是一个语法问题，我找不到合适的文档。

Answer 1

您已经交换了声明和设置。以相反的方式将参数传递给 Timer()。

Answer 2

尝试一下：

def tmp():
    import maxcut as mc 
    import networkx as nx 
    G = nx.complete_graph(3)
    mc.largest_eigenvector(G)

t = timeit.Timer(s, 'tmp()')

以下方法也可能有效：

t = timeit.Timer(setup=s, stmt='mc.largest_eigenvector(G)')