Python流式输出实战:详解实用技巧
import sys
# 定义一个简单的装饰器,用于计时函数执行时间
def timer(func):
def wrapper(*args, **kwargs):
start = time.time()
result = func(*args, **kwargs)
end = time.time()
print(f"{func.__name__} 运行时间: {end - start:.2f}秒")
return result
return wrapper
# 定义一个简单的生成器函数,用于模拟大量数据的处理
@timer
def process_data(lines):
for line in lines:
# 这里只是示例,实际应用中可以进行复杂的处理
yield line.upper()
# 在Python 3中,可以直接使用sys.stdout来进行流式输出
@timer
def stream_output(lines):
for line in lines:
sys.stdout.write(line)
sys.stdout.flush()
# 使用示例
data_lines = ["行1\n", "行2\n", "行3\n"]
processed_lines = process_data(data_lines)
for line in processed_lines:
print(line, end='') # 此处不使用list()是因为要保持流式输出
# 输出文件对象,可以是任何实现了write()方法的对象
output_file = sys.stdout
stream_output(process_data(data_lines))
这段代码首先定义了一个装饰器timer
来计算函数执行时间,然后定义了一个生成器函数process_data
来模拟数据处理。在stream_output
函数中,使用sys.stdout
进行流式输出。最后,给出了一个使用示例,展示了如何处理一系列数据并进行流式输出。
评论已关闭