Python学习打卡--day43(基础练习:装饰器)

当一个函数中,不同逻辑混杂在一起的时候,程序的可读性会大打折扣。这个时候,可以考虑用一种叫做“装饰器”的东西来重新整理代码。

# 判断一个数是不是质数
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
# 打印10000以内的素数,统计运行耗时
def prime_num():
    start_time = time.time()
    for i in range(2, 10000):
        if is_prime(i):
            print(i)
    end_time = time.time()
    print("运行耗时:{}".format(end_time-start_time))
prime_num()

prime_num中既有打印质素,又有打印运行耗时,可读性差;为了把功能分开,引入装饰器

import time
# 装饰器,打印运行时间
def display_time(func):
    def wrapper():
        start_time = time.time()
        func()
        end_time = time.time()
        print("运行耗时:{}".format(end_time - start_time))
    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
# 打印2000以内的素数
@display_time
def prime_num():
    for i in range(2, 10000):
        if is_prime(i):
            print(i)
prime_num()

如果求10000以内的质数,程序修改为

import time
# 装饰器,打印运行时间
def display_time(func):
    def wrapper():
        start_time = time.time()
        func()
        end_time = time.time()
        print("运行耗时:{}".format(end_time - start_time))
    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
# 打印2000以内的素数
@display_time
def prime_num():
    count = 0
    for i in range(2, 10000):
        if is_prime(i):
            count = count + 1
    print(count)
    return count
count = prime_num()
print(count)
返回结果:
1229
运行耗时:0.428713321685791
None

count并没有输出来,因为函数返回的值并没有返回到wrapper中,在装饰器中加入返回值

import time
# 装饰器,打印运行时间
def display_time(func):
    def wrapper():
        start_time = time.time()
        result = func()
        end_time = time.time()
        print("运行耗时:{}".format(end_time - start_time))
        return result
    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
# 打印2000以内的素数
@display_time
def prime_num():
    count = 0
    for i in range(2, 10000):
        if is_prime(i):
            count = count + 1
    print(count)
    return count
count = prime_num()
print(count)
打印结果:
1229
运行耗时:0.6667366027832031
1229

若要计算2~X之间的质数,函数带有参数,那么如何带到装饰器wrapper中?
*args表示可传入多个参数

import time
# 装饰器,打印运行时间
def display_time(func):
    def wrapper(*args):  # 可多个参数
        start_time = time.time()
        result = func(*args)
        end_time = time.time()
        print("运行耗时:{:.4}".format(end_time - start_time))
        return result

    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
# 打印2000以内的素数
@display_time
def prime_num(maxnum):
    count = 0
    for i in range(2, maxnum):
        if is_prime(i):
            count = count + 1
    print(count)
    return count
count = prime_num(10000)
print(count)
运行结果
1229
运行耗时:0.4028
1229
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值