為什么 dict 類型的變量內容不能腌制？

Question

我必須使用 numba 的 dict (numba.typed.typeddict.Dict) 類型來定義字典：

@njit
def init_numba_dict():
    x = numpy.array([0], dtype=numpy.int32)
    return {"01234-5": x[0], "0-1": x[0]}

myDict = init_numba_dict()

現在我需要將它保存到一個文件中。 我認為以下代碼可以工作：

import pickle
filehandler = open("file.dat", 'wb') 
pickle.dump(myDict , filehandler, pickle.HIGHEST_PROTOCOL)
filehandler.close()

我收到此錯誤：“TypeError：無法腌制 _nrt_python._MemInfo 對象”

有什么提示嗎？ 如何將 numba.typed.typeddict.Dict 的內容保存和恢復到文件中？

Answer 1

好問題。 我剛剛遇到同樣的錯誤並找到了解決方案。 發生錯誤是因為 numba（C 語言）返回一個指針，而不是數組或字典中的 python 之類的值。 另一方面，pickle 要求輸入 python object 作為輸入。 因此，您需要做的就是在使用 dict() function 酸洗之前將 numba dict 更改為 python dict object。

import pickle
filehandler = open("file.dat", 'wb') 
pickle.dump(dict(myDict), filehandler, pickle.HIGHEST_PROTOCOL)
filehandler.close()