Posso armazenar um dicionário usando o np.savez? Os resultados são surpreendentes (pelo menos para mim) e não consigo encontrar uma maneira de recuperar meus dados por chave.

In [1]: a = {'0': {'A': array([1,2,3]), 'B': array([4,5,6])}}
In [2]: a
Out[2]: {'0': {'A': array([1, 2, 3]), 'B': array([4, 5, 6])}}

In [3]: np.savez('model.npz', **a)
In [4]: a = np.load('model.npz')
In [5]: a
Out[5]: <numpy.lib.npyio.NpzFile at 0x7fc9f8acaad0>

In [6]: a['0']
Out[6]: array({'B': array([4, 5, 6]), 'A': array([1, 2, 3])}, dtype=object)

In [7]: a['0']['B']
---------------------------------------------------------------------------
ValueError                                Traceback (most recent call last)
<ipython-input-16-c916b98771c9> in <module>()
----> 1 a['0']['B']

ValueError: field named B not found

In [8]: dict(a['0'])
---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
<ipython-input-17-d06b11e8a048> in <module>()
----> 1 dict(a['0'])

TypeError: iteration over a 0-d array

Eu não entendo exatamente o que está acontecendo. Parece que meus dados se tornam um dicionário dentro de uma matriz 0-dimensional, deixando-me sem meios de recuperar meus dados por chave. Ou eu estou esquecendo de alguma coisa?

Então, minhas perguntas são:

o que acontece aqui? Se ainda posso acessar meus dados por chave, como?Qual é a melhor maneira de armazenar dados desse tipo? (um ditado com str como chave e outros ditados como valor)

Obrigado!