在Python中生成包含4列的1 GB文件

import random
import string

def generate_file(file_size):
    file_name = "output.txt"
    column_count = 4
    column_separator = "\t"
    row_separator = "\n"
    buffer_size = 1024 * 1024  # 1 MB buffer size

    with open(file_name, "w") as file:
        while file.tell() < file_size:
            row = []
            for _ in range(column_count):
                random_string = ''.join(random.choices(string.ascii_letters + string.digits, k=10))
                row.append(random_string)
            file.write(column_separator.join(row) + row_separator)
            file.flush()
            if file.tell() % buffer_size == 0:
                print(f"{file.tell() / (1024 * 1024)} MB written")

    print("File generation complete.")

generate_file(1024 * 1024 * 1024)  # 1 GB file size