当前位置：首页 > news >正文

python生成器

news 来源：原创 2025/6/9 7:07:45

什么是生成器

根据开发人员指定的规则循环生成数据，当条件不成立时则生成数据结束。数据不是一次性全部生产出来，而是使用一个，再生成一个，可以节约大量内存。

创建生成器的方式

生成器推导式
yield 关键字

生成器推导式

生成器相关函数

next 函数获取生成器中的下一个值
for 循环遍历生成器中的每一个值

# 创建生成器
# 注意1：括号() 代表这是一个生成器，不是元组
# 注意2：括号()里面写的是数据的生成规则，返回一个对象，对象内不是存储的数据，而是生成数据的规则my_generator = (i * 2 for i in range(5))print(my_generator)  # 输出：<generator object <genexpr> at 0x000001B8ABC3B9F0>，说明这是一个生成器对象，不是存储数据的容器# next 获取生成器下一个值
# value = next(my_generator)
# print(value)  # 输出：0，说明生成器每次调用next()方法时，会根据生成规则生成下一个值# for循环获取生成器的值
for value in my_generator:print(value)  # 输出：0 2 4 6 8，说明for循环会自动调用next()方法获取生成器下一个值，直到生成器没有值为止

`yield` 生成器

只要在def函数里面看到有yield 关键字，那么就是生成器

def my_generator(n):for i in range(n):print("开始生成...")yield iprint("生成结束...")if __name__ == '__main__':gen = my_generator(5)  # 创建生成器对象，不会执行生成规则，只有在调用next()方法时才会执行生成规则#获取生成器中下一个值# result = next(gen)  # print(result)  # while True:#     try:#         result = next(gen)  #         print(result)  #     except StopIteration as e:  #         break# for循环遍历生成器，for循环内部自动处理了停止迭代异常，使用起来更加方便for value in gen:print(value)

生成器高级应用场景

很多模型都是一个批次一个批次的给模型喂数据，来训练模型的

import mathdef dataset_loader(bath_size):with open('./jaychou_lyrics.txt', 'r', encoding='utf-8') as f:lines = f.readlines()# 统计有多少条歌词lyrics_num = len(lines)  # 歌词数量# 计算共有多少个批次 match.ceil 向上取整,math.floor 向下取整batch_num = math.ceil(lyrics_num / bath_size)  # 批次数量# 遍历每个批次for i in range(batch_num):yeild lines[i * bath_size: (i + 1) * bath_size]  # 每次返回一个批次的数据if __name__ == '__main__':# 调用函数，传入批次大小for batch in dataset_loader(8):  # 批次大小为8print(batch)  # 输出每个批次的数据

property属性

property属性介绍

负责把一个方法当做属性进行使用，这样做可以简化代码使用
定义property属性有两个方法
- 装饰器方式
- 类属性方式

装饰器方式

@property 修饰获取值的方法
@方法名.setter 修饰设置值的方法

class Person(object):def __init__(self):self._age = 0# 获取属性@propertydef age(self):return self._age# 设置属性@age.setterdef age(self, value):if value < 0:raise ValueError("年龄不能为负数")self._age = valueif __name__ == '__main__':person = Person()print(person.age)  # 使用 @property 装饰器获取年龄person.age = 20  # 使用 @age.setter 装饰器设置年龄print(person.age)  # 使用 @property 装饰器获取年龄

类属性方式

类属性 = property(获取值方法，设置值方法)

class Person(object):def __init__(self):self._age = 0def get_age(self):return self._agedef set_age(self, value):if value < 0:raise ValueError("年龄不能为负数")self._age = value# 使用 property 装饰器定义属性age = property(get_age, set_age)if __name__ == '__main__':person = Person()print(person.age) person.age = 20  print(person.age)