用python读取csv文件,然后将目标文件夹中的图片分类到相应文件夹中
时间: 2024-04-16 07:24:28 浏览: 130
使用python将图片按标签分入不同文件夹的方法
5星 · 资源好评率100%
要读取CSV文件并将目标文件夹中的图片分类到相应文件夹中,您可以使用Python的csv模块和shutil模块来完成任务。以下是一个示例代码:
```python
import csv
import os
import shutil
def classify_images(csv_file, source_folder, destination_folder):
with open(csv_file, 'r') as file:
reader = csv.reader(file)
next(reader) # 跳过标题行
for row in reader:
filename = row[0] # 假设CSV中的第一列是文件名
category = row[1] # 假设CSV中的第二列是分类
source_path = os.path.join(source_folder, filename)
destination_path = os.path.join(destination_folder, category, filename)
if os.path.exists(source_path):
if not os.path.exists(os.path.join(destination_folder, category)):
os.makedirs(os.path.join(destination_folder, category))
shutil.move(source_path, destination_path)
print(f"Moved {filename} to {os.path.join(destination_folder, category)}")
else:
print(f"File {filename} does not exist")
# 读取CSV文件并将图片分类到相应文件夹
csv_file = 'data.csv' # 替换为您的CSV文件路径
source_folder = 'source' # 替换为您的源文件夹路径
destination_folder = 'destination' # 替换为您的目标文件夹路径
classify_images(csv_file, source_folder, destination_folder)
```
在上述代码中,您需要将`csv_file`变量替换为您的CSV文件路径,`source_folder`变量替换为您的源文件夹路径,`destination_folder`变量替换为您的目标文件夹路径。代码将遍历CSV文件中的每一行,将源文件夹中的图片移动到相应分类的文件夹中。
请注意,代码假设CSV文件的第一列是图片文件名,第二列是分类。您可以根据实际情况进行修改。此外,代码还会检查源文件是否存在,如果不存在,则会输出相应的警告信息。
阅读全文