内容有空补上
Python字典可以存储ndarray数组类型,但是由dict序列化为JSON文件时,无法序列化ndarray类型,为了实现读写numpy数组,需要重写JSONEncoder的default方法,基本原理也就是先把ndarray转化为list,然后后续读取再从list转化为ndarray。
import json import numpy as np # 重写JSONEncoder中的default方法 class NumpyArrayEncoder(json.JSONEncoder): def default(self, obj): if isinstance(obj, np.ndarray): return obj.tolist() return json.JSONEncoder.default(self, obj) arr_1 = np.random.rand(3, 3) arr_2 = np.random.rand(2) data = { 'arr1': arr_1, 'arr2': arr_2 } # 将存有ndarray的字典转化为json类型 data_json = json.dumps(data, cls=NumpyArrayEncoder) print(f'data_json:{data_json}') # 写入json文件 with open('array_json.json', 'w') as f: json.dump(data, f, cls=NumpyArrayEncoder) # 直接从json类型转化为字典 data_loads = json.loads(data_json) print(f'data_loads:{data_loads}') # 从json文件读取numpy数组 with open('array_json.json', 'r') as f: data_read = json.load(f) arr1_read = np.asarray(data_read['arr1']) arr2_read = np.asarray(data_read['arr2']) print(f'arr1_read:{arr1_read}\narr2_read:{arr2_read}')
参考网址