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

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


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39

参考网址