加载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