利用json文件将文件夹的图片分类
时间: 2024-05-06 16:19:48 浏览: 142
把文件夹中的图片分类并记录在json文件中,可以按照以下步骤进行:
1. 打开文件夹,获取文件夹中所有的图片文件。
2. 遍历所有的图片文件,根据图片的类型(如人物、动物、风景等)进行分类。
3. 将分类后的图片文件路径记录在对应的json文件中。可以使用Python中的json模块对json文件进行读写操作。例如,可以创建一个字典,以图片类型为键,对应的图片文件路径列表为值,然后使用json.dump()函数将字典写入json文件中。
4. 最后,保存json文件并关闭所有文件。
以下是一个示例代码,用于将文件夹中的图片分类并保存在json文件中:
``` python
import os
import json
# 文件夹路径
folder_path = "path/to/folder"
# 图片类型列表
types = ["people", "animal", "scenery"]
# 分类字典
data = {t: [] for t in types}
# 遍历文件夹中的所有文件
for filename in os.listdir(folder_path):
# 获取文件路径
filepath = os.path.join(folder_path, filename)
# 判断是否为图片文件
if os.path.isfile(filepath) and filename.endswith(".jpg"):
# 根据文件名判断图片类型
for t in types:
if t in filename.lower():
data[t].append(filepath)
# 写入json文件
with open("image_data.json", "w") as f:
json.dump(data, f)
print("分类完成并保存在image_data.json文件中。")
```
在上面的代码中,我们首先定义了一个文件夹路径和一个图片类型列表。然后创建了一个分类字典,以图片类型为键,对应的图片文件路径列表为空列表为值。通过遍历文件夹中的所有文件,判断是否为图片文件,并根据文件名判断图片类型,将文件路径添加到相应的列表中。最后,使用json.dump()函数将分类字典写入json文件中。
阅读全文