使用sys.stdout.flush()函数提升代码性能和效率的技巧

sys.stdout.flush()函数在Python中的主要作用是刷新标准输出，并将缓冲区中的内容立即输出到终端。它可以用于提升代码的性能和效率，特别是在需要实时输出结果或处理大量输出时。

在默认情况下，Python的print语句会将内容输出到缓冲区中，并在合适的时机自动刷新缓冲区，将内容输出到终端。这种自动刷新缓冲区的机制使得输出看起来更加平滑，但当处理大量输出时会带来性能问题。

当我们需要实时获取输出结果或处理大量输出时，可以使用sys.stdout.flush()函数手动刷新缓冲区，将内容立即输出到终端。这样可以减少缓冲区的开销，提高输出效率和性能。

下面以两个示例来演示sys.stdout.flush()函数的使用：

示例一：实时输出结果

假设我们有一个耗时很长的任务，需要实时输出进度信息。使用print函数输出结果，由于默认的缓冲机制，进度信息不会立即显示，直到任务执行完毕才一起显示。

import time

def long_running_task():
    for i in range(5):
        time.sleep(1)
        print(f"Progress: {i + 1} out of 5")
    print("Task completed!")

long_running_task()

上述代码会在任务结束后一次性输出所有进度信息，结果如下：

Progress: 1 out of 5
Progress: 2 out of 5
Progress: 3 out of 5
Progress: 4 out of 5
Progress: 5 out of 5
Task completed!

如果希望实时输出进度信息，可以在每次print语句后调用sys.stdout.flush()函数刷新缓冲区，将内容立即输出到终端。修改后的代码如下：

import time
import sys

def long_running_task():
    for i in range(5):
        time.sleep(1)
        print(f"Progress: {i + 1} out of 5")
        sys.stdout.flush()
    print("Task completed!")

long_running_task()

修改后的代码会在每次循环中实时输出进度信息到终端，结果如下：

Progress: 1 out of 5
Progress: 2 out of 5
Progress: 3 out of 5
Progress: 4 out of 5
Progress: 5 out of 5
Task completed!

示例二：处理大量输出

假设我们需要处理一个非常大的列表，并将每个元素的平方输出。使用print函数处理大量输出时，由于默认的缓冲机制，会导致在处理完成之前无法看到任何输出。

def process_large_list():
    large_list = range(1000000)
    for i in large_list:
        print(i ** 2)
    print("Processing completed!")

process_large_list()

上述代码会在处理完整个列表后才输出结果，结果如下：

Processing completed!

如果希望实时输出每个元素的平方，可以在每次print语句后调用sys.stdout.flush()函数刷新缓冲区，将内容立即输出到终端。修改后的代码如下：

import sys

def process_large_list():
    large_list = range(1000000)
    for i in large_list:
        print(i ** 2)
        sys.stdout.flush()
    print("Processing completed!")

process_large_list()

修改后的代码会在处理每个元素时实时输出结果到终端，结果如下：

0
1
4
9
16
...
999996001
999998000
Processing completed!

通过使用sys.stdout.flush()函数手动刷新缓冲区，我们可以实时获取输出结果或处理大量输出，从而提升代码的性能和效率。