numpy中的savez（）同时保存数组、字符串、字典数据

原文链接：https://blog.csdn.net/lilong117194/article/details/83589510

python使用numpy中的savez（）保存非数组数据到本地。

报错：

IndexError: only integers, slices (`:`), ellipsis (`...`), numpy.newaxis (`None`) and integer or boolean arrays are valid indices
1

看下面的代码实例：

import numpy as np


# 简单的数组保存
arr = np.array([[1, 2], [3, 4]])
np.save('/Users/lilong/Desktop/mm.npy',arr)
arr_load=np.load('/Users/lilong/Desktop/mm.npy')
print(arr_load,'\n\n',arr_load[0])


# 非结构化数据保存
str_ = 'abc'
arr_ = np.array([[1, 2], [3, 4]])
dict_ = {'a' : 1, 'b': 2}
np.savez('/Users/lilong/Desktop/nn.npz', st= str_, ar = arr_, dic= dict_)


data = np.load('/Users/lilong/Desktop/nn.npz')
print('....\n',data['st'],'\n',data['st'][()])
print('....\n',data['ar'],'....\n',data['ar'][0],'\n\n',data['ar'][()][0])
print('....\n',data['dic'],'\n\n',data['dic'][()])
#print('....\n',data['dic']['a'])  # 报错
print('....\n',data['dic'][()]['a'])

运行结果：

[[1 2]
 [3 4]] 

 [1 2]
....
 abc 
 abc
....
 [[1 2]
 [3 4]] ....
 [1 2] 

 [1 2]
....
 {'a': 1, 'b': 2} 

 {'a': 1, 'b': 2}
....
 1