加載Python Pickle（.pkl）文件

Question

我正在嘗試加載.py文件所在目錄中的.pkl文件。 以下是我的代碼：

import os

def load_var(var_name):
    fid = open(os.path.join((var_name, '.pkl')))
    data = pickle.load(fid)
    fid.close()
    return data

def main():
    data = load_var('myfilename')

if __name__ == '__main__':
    main()

我繼續遇到錯誤：

fid = open(os.path.join((var_name, '.pkl')))
TypeError: coercing to Unicode: need string or buffer, tuple found

是否有解決此錯誤的簡便方法？

Answer 1

os.path.join的輸入不應為元組。 也就是說，該命令應顯示為：

fid = open(os.path.join(var_name, '.pkl'))

Answer 2

C。 皮革是正確的。 卸下多余的括號（）。 還要注意，您正在使用os.path.join（）來處理文件名和擴展名？

該輸出將是var_name / .pkl，不確定是否正是您所期望的。

>>> filename = "mytestfile"
>>> mypath = os.path.join(filename, ".pkl")
>>> 
>>> print mypath
mytestfile/.pkl