Python:从.txt、字符串键和数组类型值读取字典

Question

我已将字典保存到 txt 文件中

f = open("dict_imageData.txt","a")
f.write( str(dict_imageData) + "\n" )
f.close()

保存的txt文件包含

{'002_S_0559': array([ 0.,  0.,  0., ...,  0.,  0.,  0.],dtype=float32)}
{'002_S_1070': array([ 0.,  0.,  0., ...,  0.,  0.,  0.], dtype=float32)}
{'023_S_0604': array([ 0.,  0.,  0., ...,  0.,  0.,  0.], dtype=float32)}

我在加载此字典以及拆分键和值时遇到问题。我尝试了 split 和 eval 但没有成功，因为最后有一个 dtype 语句。

有什么建议吗？

Answer 1

感谢您的评论，是的，我知道使用 JSON 和 pickle，但我对 .txt 文件更感兴趣。我解决了我的问题并找到了如下解决方案:

首先，我将 MyEncoder 类定义为:

class MyEncoder(json.JSONEncoder):
def default(self, obj):
    if isinstance(obj, np.integer):
        return int(obj)
    elif isinstance(obj, np.floating):
        return float(obj)
    elif isinstance(obj, np.ndarray):
        return obj.tolist()
    else:
        return super(MyEncoder, self).default(obj)

由于我的键是字符串，值是数组，因此在写入键值对之前，我将值转储为

dict_imageData [key] = json.dumps(np.float32(np.array(values)), cls = MyEncoder)

现在，我将字典写入文件中

with open('dict_imageData.txt', 'a') as file:
    file.write(json.dumps(dict_imageData))
    file.write("\n")

为了从 .txt 文件读回字典，我使用 eval

with open('dict_imageData.txt','r') as inf:
    lineno = 0
    for line in inf:  
        lineno = lineno + 1 
        print("line number in the file: ", lineno)
        dicts_from_file = dict()
        dicts_from_file = (eval(line))
        for key,value in dicts_from_file.items():
            print("key: ", key, "value: ", value)