Python读写JSON的操作及JSON与Numpy数组的存储与读取

JSON基本操作

内容有空补上

JSON与Numpy数组的读写操作

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}')


参考网址

上一篇:【学习笔记】Python - NumPy


下一篇:成为华尔街金融巨鳄第一课:学会使用NumPy