在ubuntu系统把txt文件夹中的所有文件名称提取到txt文件按顺序排序

import os

def extract_filenames_from_folder(input_folder, output_txt):
    try:
        # 检查输入文件夹是否存在
        if not os.path.exists(input_folder):
            print(f"Error: Input folder '{input_folder}' does not exist.")
            return

        # 获取文件夹中所有文件的文件名
        filenames = [file for file in os.listdir(input_folder) if os.path.isfile(os.path.join(input_folder, file))]

        # 检查是否存在文件
        if not filenames:
            print(f"No files found in '{input_folder}'.")
            return

        # 对文件名进行自然排序
        filenames.sort(key=lambda x: int(os.path.splitext(x)[0]))

        # 将文件名写入输出txt文件（去掉文件后缀）
        with open(output_txt, 'w') as output:
            for filename in filenames:
                output.write(f"{os.path.splitext(filename)[0]}\n")

        print(f"File names extracted, sorted, and saved to '{output_txt}'.")
    except Exception as e:
        print(f"An error occurred: {e}")

# 调用函数，传入输入文件夹和输出文件的路径
extract_filenames_from_folder('/home/zhang/yolov5_obb_master/train_bottle/train/labels', '/home/zhang/yolov5_obb_master/train_bottle/train/imgnamefile.txt')