要按照CSV文件中的名称对图像进行筛选,可以使用以下步骤:
import csv
csv_file = 'path/to/file.csv'
with open(csv_file, 'r') as file:
csv_reader = csv.reader(file)
image_names = [row[0] for row in csv_reader]
import os
image_folder = 'path/to/images'
images = []
for filename in os.listdir(image_folder):
if filename.endswith('.jpg') or filename.endswith('.png'):
images.append(filename)
selected_images = []
for image_name in image_names:
if image_name in images:
selected_images.append(image_name)
现在,selected_images列表中包含了与CSV文件中的名称匹配的图像文件名称。
请注意,以上代码示例假设CSV文件中的图像名称与文件夹中的图像文件名称完全匹配。如果CSV文件中的图像名称仅是图像文件名称的一部分,可以使用字符串匹配算法(如模糊匹配或正则表达式)来进行筛选。