python按照一定的行数将文件切分成多个文件

import sys

def split_file(input_file, lines_per_file):
    with open(input_file, 'r') as f:
        current_file_index = 1
        current_line = 0
        output_file = None
        for line in f:
            if current_line % lines_per_file == 0:
                if output_file:
                    output_file.close()
                output_filename = "{}_{:03d}.txt".format(input_file, current_file_index)
                output_file = open(output_filename, 'w')
                current_file_index += 1
            output_file.write(line)
            current_line += 1
        if output_file:
            output_file.close()

if __name__ == "__main__":
    input_file = sys.argv[1]  # 输入文件名
    lines_per_file = int(sys.argv[2])  # 每个文件的行数
    split_file(input_file, lines_per_file)