如何根据第一个字母后的数字对文本文件进行排序

import re

def extract_number(line):
    match = re.search(r'([a-zA-Z])(\d+)', line)
    if match:
        return int(match.group(2))
    return float('inf')  # 如果没有找到数字，返回无穷大

def sort_file(input_file, output_file):
    with open(input_file, 'r') as file:
        lines = file.readlines()
    
    sorted_lines = sorted(lines, key=extract_number)
    
    with open(output_file, 'w') as file:
        file.writelines(sorted_lines)

# 使用示例
input_file = 'input.txt'
output_file = 'sorted_output.txt'
sort_file(input_file, output_file)