如何正确创建pickle文件?

时间:2017-12-26 21:48:16

标签: python numpy pickle contextmanager

我正在尝试创建一个pickle文件。 我使用了以下代码来完成它:

def pickler(input_nparray):
    with open('/Users/username/Desktop/pklfilename.pkl', 'wb+') as f:
        pickle.dump(input_nparray, f)

这种方法非常有效。这样做是在numpy数组中将内容存储到pklfilename.pkl中。

但问题是我必须指定文件名。我尝试以下列方式做,但失败了。

def pickler(input_nparray, pklfilename):
    with open('/Users/swaghccc/Desktop/' + pklfilename, 'wb+') as f:
        pickle.dump(input_nparray, f)

pickler(input_nparray, 'file1.pkl')

有人可以告诉我正确的选择吗?

1 个答案:

答案 0 :(得分:0)

第一个和第二个示例的用户名不同。将其更改为username并使用文件名pklfilename.pkl,看看您是否在第二个示例中以文件结尾。

另一个提示是将文件名保存在一个变量中,打印它以试图找出发生了什么。