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')
在ubuntu系统把txt文件夹中的所有文件名称提取到txt文件按顺序排序
于 2024-01-17 18:45:03 首次发布
![](https://img-home.csdnimg.cn/images/20240711042549.png)