python使用numpy保存字典格式的数据

这里写一个保存用的工具函数：

def save_dict_by_numpy(filename, dict_vale):
    if not (os.path.exists(os.path.dirname(filename))):
        os.mkdir(os.path.dirname(filename))
    np.save(filename, dict_vale)

保存时用：np.save(保存文件名, dict的变量)
加载时用：np.load(保存文件名, allow_pickle=True)，这会返回一个object的ndarry格式的数据，后面再加一个.item()就可以变为字典格式

示例代码

import os
import numpy as np


def save_dict_by_numpy(filename, dict_vale):
    if not (os.path.exists(os.path.dirname(filename))):
        os.mkdir(os.path.dirname(filename))
    np.save(filename, dict_vale)


my_dict = {"name": "小明", "age": 12}
output_filename = "save_dir/my_dict.npy"

# 保存
save_dict_by_numpy(output_filename, my_dict)

# 加载
my_dict2 = np.load(output_filename, allow_pickle=True).item()

print(my_dict, my_dict2)