Python读取图片名称并写入txt

在Python中，你可以使用os模块来获取图片文件的名称，并使用内置的open()函数将其写入文本文件。首先，你需要遍历包含图片的目录，然后使用os.path.basename()函数获取每个文件的名称。下面是一个简单的示例：

import os

def get_image_names(directory):
    image_names = []
    for filename in os.listdir(directory):
        if filename.endswith(('.jpg', '.jpeg', '.png', '.bmp')):
            # 只保留图片文件
            image_names.append(os.path.join(directory, filename))
    return image_names

def write_image_names_to_txt(image_names, output_file):
    with open(output_file, 'w') as file:
        for name in image_names:
            file.write(name + '\n')

# 使用方法
image_dir = r'C:\path\to\your\images'  # 替换为实际图片目录
output_txt = 'image_names.txt'
image_names = get_image_names(image_dir)
write_image_names_to_txt(image_names, output_txt)

这段代码首先定义了一个get_image_names函数，它会在指定目录下查找所有图片文件（这里只检查了常见的几种格式），并将它们的完整路径添加到列表中。然后，write_image_names_to_txt函数接收这个列表以及一个输出文件名，逐行将图片名称写入到文本文件中。