numpy.ndarrayまたはネストされたリスト構成をJSONとして保存します。
class NumpyEncoder(json.JSONEncoder):
def default(self, obj):
if isinstance(obj, np.ndarray):
return obj.tolist()
return json.JSONEncoder.default(self, obj)
a = np.array([[1, 2, 3], [4, 5, 6]])
print(a.shape)
json_dump = json.dumps({'a': a, 'aa': [2, (2, 3, 4), a], 'bb': [2]}, cls=NumpyEncoder)
print(json_dump)
出力されます:
(2, 3)
{"a": [[1, 2, 3], [4, 5, 6]], "aa": [2, [2, 3, 4], [[1, 2, 3], [4, 5, 6]]], "bb": [2]}
JSONから復元するには:
json_load = json.loads(json_dump)
a_restored = np.asarray(json_load["a"])
print(a_restored)
print(a_restored.shape)
出力されます:
[[1 2 3]
[4 5 6]]
(2, 3)
json
モジュールを使用して派手な配列をダンプしようとしていることを意味します。しかし、処理方法を知ってnumpy.ndarray
いるタイプではありませんjson
。独自list(your_array)
のシリアライザーを作成するか、(より簡単に)jsonを作成しているものに渡す必要があります。