Python装饰器打印“无”

Question

import random
import time

def timeit(func):

    def wrapper(*args, **kwargs):
        start = time.time()
        func(*args, **kwargs)
        print ("{} took {} secs \n".format(func.__name__, time.time() - start))
    return wrapper


@timeit
def insort(l):
    """ Uses insetion sort for sorting list """

    for i in range(1, len(l)): 
        temp = l[i]
        j = i
        while j > 0 and l[j - 1] > temp:
            l[j] = l[j - 1]
            j -= 1
            l[j] = temp
    return l


@timeit
def bublesort(l):
    """ Uses insetion sort for sorting list """

    for i in range(len(l)-1, 0, -1):
        for j in range(i):
            if l[j] > l[j+1]:
                l[j], l[j+1] = l[j+1], l[j]
    return l


x =random.sample(range(2000), 1000) 
print (insort(x[:]))
print (bublesort(x[:]))

上面的代码输出：

insort took 0.0629999637604 secs 

None
bublesort took 0.104000091553 secs 

None

为什么在所有结果后都不打印，我该如何防止呢？ 我也是装饰器的新手，所以我想知道是否有更好的方法来计时我的代码。

Answer 1

您的包装器将替换原始函数，它将忽略返回值并返回None 。

进行以下更改：

def timeit(func):

    def wrapper(*args, **kwargs):
        start = time.time()
        ret = func(*args, **kwargs)
        print ("{} took {} secs \n".format(func.__name__, time.time() - start))
        return ret
    return wrapper

这将使您的装饰器正常工作。 如果要避免打印返回值，请更改

print (insort(x[:]))
print (bublesort(x[:]))

只是

insort(x[:])
bublesort(x[:])

通常，如果您不想打印某些东西，请不要将其传递给print 。

Answer 2

为什么在所有结果后都不打印，我该如何防止呢？ 我也是装饰器的新手，所以我想知道是否有更好的方法来计时我的代码。

由于以下原因，它正在打印包装器的返回值（无）：

print (insort(x[:]))
print (bublesort(x[:]))

您正在告诉解释器打印这些函数的返回值。 如果您不想看到“ None ，则应调用以下函数：

insort(x[:])
bublesort(x[:])