标签：Python Numpy json read JSON np data ndarray

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,json,read,JSON,np,data,ndarray
来源： https://blog.csdn.net/weixin_43167168/article/details/121129776

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。