装饰器基础

最新推荐文章于 2024-10-01 05:04:32 发布

笔下万码生谋略

最新推荐文章于 2024-10-01 05:04:32 发布

阅读量61

点赞数

分类专栏： Python 文章标签： python 开发语言

本文链接：https://blog.csdn.net/weixin_47707533/article/details/125547891

版权

Python 专栏收录该内容

3 篇文章 0 订阅

订阅专栏

本文介绍了Python装饰器的三种常见用法：无返回值、有返回值以及处理带参数的函数。通过`@display_time`装饰器，展示了如何计算并打印函数的运行时间和结果，以及如何在装饰器中处理被装饰函数的参数。示例包括计算素数和统计一定范围内素数的数量。

摘要由CSDN通过智能技术生成

代码如下（示例）：

import time
# 例1 无返回值
def display_time(func):
    def wrapper():
        t1 = time.time()
        func()
        t2 = time.time()
        print(f'运行时间：{t2-t1:.4}s')
    return wrapper

def is_prime(num):
    if num < 2:
        return False
    elif num == 2:
        return True
    else:
        for i in range(2, num):
            if num % i == 0:
                return False
        return True

@display_time
def prime_nums():
    for i in range(2, 10000):
        if is_prime(i):
            print(i)
prime_nums()


# 例2 有返回值,需要在装饰器返回出来
def display_time(func):
    def wrapper():
        t1 = time.time()
        ret = func()
        t2 = time.time()
        print(ret)
        print(f'运行时间：{t2-t1:.4}s')
    return wrapper

def is_prime(num):
    if num < 2:
        return False
    elif num == 2:
        return True
    else:
        for i in range(2, num):
            if num % i == 0:
                return False
        return True

@display_time
def count_prime_nums():
    count = 0
    for i in range(2, 10000):
        if is_prime(i):
            count += 1
    return count
count_prime_nums()

 
# 例3 被装饰函数需传参
def display_time(func):
    def wrapper(*args):
        t1 = time.time()
        ret = func(*args)
        t2 = time.time()
        print(ret)
        print(f'运行时间：{t2-t1:.4}s')
    return wrapper

def is_prime(num):
    if num < 2:
        return False
    elif num == 2:
        return True
    else:
        for i in range(2, num):
            if num % i == 0:
                return False
        return True

@display_time
def count_prime_nums(maxnum):
    count = 0
    for i in range(2, maxnum):
        if is_prime(i):
            count += 1
    return count
count_prime_nums(5000)