我们在处理json文件或者根据json文件生成xml文件时,有时需要提取label值(其他雷同)。
import os
import json
# 主文件夹路径
folder_path = r'F:\code_demo\测试二/'
# 创建一个集合来存储标签
labels_set = set()
json_files = [file for file in os.listdir(folder_path) if file.endswith('.json')]
if json_files:
# # 从JSON文件中随机选择一个
# random_json_file = random.choice(json_files)
# json_file_path = os.path.join(folder_path, random_json_file)
for json_file in json_files:
json_path = os.path.join(folder_path, json_file)
# 打开JSON文件并解析内容
with open(json_path, 'r', encoding="utf-8") as json_file:
# json_data = json.load(json_file)
data = json.load(json_file)
for shape in data["shapes"]:
label = shape["label"]
if isinstance(label, str):
print(f"JSON文件:{json_path},标签:{label}")
labels_set.add(label)
# 打印集合中的标签
print('--------------------')
for label in labels_set:
print(label)