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

2021/11/3 20:40:03

本文主要是介绍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读写JSON的操作及JSON与Numpy数组的存储与读取的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!


扫一扫关注最新编程教程