【Python Numpy】关于用numpy object对象类型 load savez读写Python的字典类型dict

问题描述

最近在用Python写一些人工智能相关的代码，然后想把神经网络，数据和其它的一些参数保存了下来。这里就想到了用Numpy压缩保存命令savez，但是之前只有存储过数组，并没有试过存其它的Python类如dict等.
然而就壮了胆的尝试了一下以下的代码,

import numpy as np
np.savez('t', d1={"jia":1,"yi":2}, d2={"bing":3,"ding":4})
d=np.load('t.npz',allow_pickle=True)

输出为

>>> d['d1']
array({'jia': 1, 'yi': 2}, dtype=object)
>>> d['d2']
array({'bing': 3, 'ding': 4}, dtype=object)

当访问object对象里的数据时，遗憾的就报错了。

>>> d['d1']['jia']
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
IndexError: only integers, slices (`:`), ellipsis (`...`), numpy.newaxis (`None`) and integer or boolean arrays are valid indices

解决方法

通过一番折腾后最终发现可以这样来获取想要的字典数据,

d1 = d['d1'].reshape((-1))[0]

就是先把numpy 对象转成一个一维的数组，然后再访问便可。

>>> d1 = d['d1'].reshape((-1))[0]
>>> d1
{'jia': 1, 'yi': 2}
>>> d1['jia']
1
>>> d1['yi']
2

谢谢!