Python實現將字典內容寫入json文件
Python中有序字典和無序字典,一鍵多值字典。
Python將字典內容寫入json文件。
1、無序字典
目前瞭解三種,在Python中直接默認的是無序字典,這種不會按照你插入的順序排序,即使你對字典排序後,返回的也是一個list變量,而不是字典,倘若你將這個list字典後,又會變回無序字典。
例子如下:
import operator x = {"label": "haha", "data": 234, "score": 0.3} sorted_x = sorted(x.items(), key=operator.itemgetter(0)) print x print type(x) print sorted_x print type(sorted_x) print dict(sorted_x)
2、有序字典
如果我們想保持字典按照我們插入的順序有序怎麼辦?可以用OrderedDict來初始化字典。
例子如下:
from collections import OrderedDict x = OrderedDict() x["label"] = "haha" x["data"] = 234 x["score"] = 0.3 print x print type(x)
3、一鍵多值字典
如果我們想用一鍵多值字典怎麼辦,可以使用defaultdict,例子如下:
from collections import defaultdict video = defaultdict(list) video["label"].append("haha") video["data"].append(234) video["score"].append(0.3) video["label"].append("xixi") video["data"].append(123) video["score"].append(0.7) print video print type(video)
4、寫入json
字典內容寫入json時,需要用json.dumps將字典轉換為字符串,然後再寫入。
json也支持格式,通過參數indent可以設置縮進,如果不設置的話,則保存下來會是一行。
例子:
4.1 無縮進
from collections import defaultdict, OrderedDict import json video = defaultdict(list) video["label"].append("haha") video["data"].append(234) video["score"].append(0.3) video["label"].append("xixi") video["data"].append(123) video["score"].append(0.7) test_dict = { 'version': "1.0", 'results': video, 'explain': { 'used': True, 'details': "this is for josn test", } } json_str = json.dumps(test_dict) with open('test_data.json', 'w') as json_file: json_file.write(json_str)
4.2 有縮進
from collections import defaultdict, OrderedDict import json video = defaultdict(list) video["label"].append("haha") video["data"].append(234) video["score"].append(0.3) video["label"].append("xixi") video["data"].append(123) video["score"].append(0.7) test_dict = { 'version': "1.0", 'results': video, 'explain': { 'used': True, 'details': "this is for josn test", } } json_str = json.dumps(test_dict, indent=4) with open('test_data.json', 'w') as json_file: json_file.write(json_str)
方法補充
下面是參考上文代碼整理出的另一種實現方法,可以參考一下
""" 將整個數據集分為train和test,相應的也分別分配整個json文件 """ import os import random import json total_select_path = r"C:\Users\9ling\Desktop\YiLiuWuDataset\train\yuedongguan_select" total_json_path = r"C:\Users\9ling\Desktop\YiLiuWuDataset\train\yuedongguan.json" test_path = r"C:\Users\9ling\Desktop\YiLiuWuDataset\test\has_yiliuwu\yuedongguan_test" test_json_path = r"C:\Users\9ling\Desktop\YiLiuWuDataset\test\has_yiliuwu\yuedongguan_test\yuedongguan_test.json" train_path = r"C:\Users\9ling\Desktop\YiLiuWuDataset\train\yuedongguan" train_json_path = r"C:\Users\9ling\Desktop\YiLiuWuDataset\train\yuedongguan\yuedongguan.json" data = json.load(open(total_json_path))["labels"] # test_data = json.load(open(test_json_path))["labels"] all_select_path = os.listdir(total_select_path) all_file_path = [] # 待分配的圖片路徑 for item in all_select_path: file_path = os.path.join(total_select_path, item) all_file_path.append(file_path) # print(all_file_path) idx = [i for i in range(len(all_select_path))] random.shuffle(idx) # 在idx上改變 def copy_dir(src_path, target_path): # src_path原文件,target_path目標文件 if os.path.isdir(src_path) and os.path.isdir(target_path): filelist_src = os.listdir(src_path) for file in filelist_src: path = os.path.join(os.path.abspath(src_path), file) if os.path.isdir(path): path1 = os.path.join(os.path.abspath(target_path), file) if not os.path.exists(path1): os.mkdir(path1) copy_dir(path, path1) else: with open(path, 'rb') as read_stream: contents = read_stream.read() path1 = os.path.join(target_path, file) with open(path1, 'wb') as write_stream: write_stream.write(contents) return True else: return False test_data_dir = {"labels": []} for item in idx[:41]: with open(all_file_path[item], 'rb') as read_stream: contents = read_stream.read() path1 = os.path.join(test_path, all_file_path[item].split("\\")[-1]) # 測試集圖片的路徑 with open(path1, 'wb') as write_stream: write_stream.write(contents) for s in data: if s["filename"].split("\\")[-1] == all_file_path[item].split("\\")[-1]: test_data_dir["labels"].append(s) # print(s) json_test_str = json.dumps(test_data_dir, indent=4) with open(test_json_path, 'w') as json_file: json_file.write(json_test_str) print(test_data_dir) print(len(test_data_dir["labels"])) print("*"*30) train_data_dir = {"labels": []} for item in idx[41:]: with open(all_file_path[item], 'rb') as read_stream: contents = read_stream.read() path2 = os.path.join(train_path, all_file_path[item].split("\\")[-1]) with open(path2, 'wb') as write_stream: write_stream.write(contents) for s1 in data: if s1["filename"].split("\\")[-1] == all_file_path[item].split("\\")[-1]: train_data_dir["labels"].append(s1) json_train_str = json.dumps(train_data_dir, indent=4) with open(train_json_path, 'w') as json_file: json_file.write(json_train_str) print(train_data_dir) print(len(train_data_dir["labels"])) # print(s)
以上就是Python實現將字典內容寫入json文件的詳細內容,更多關於Python字典寫入json的資料請關註WalkonNet其它相關文章!
推薦閱讀:
- python實現MD5進行文件去重的示例代碼
- Python 如何實現批量轉換視頻音頻的采樣率
- 基於Python實現視頻去重小工具
- python 合並多個excel中同名的sheet
- 使用python對視頻文件分辨率進行分組的實例代碼