Numpy savez将我的密钥解释为文件名 - > IO错误

时间:2012-02-13 09:23:20

标签: python numpy

我使用numpy savez作为recommended来保存numpy数组。作为键我使用我加载数据的文件的名称。但似乎savez试图以某种方式使用文件名。我该怎么办?我想避免剥离其路径的文件名并结束。

>>> import numpy
>>> arrs = {'data/a.text': numpy.array([1,2]),
            'data/b.text': numpy.array([3,4]),
            'data/c.text': numpy.array([5,6])}    
>>> numpy.savez('file.npz', **arrs)
Traceback (most recent call last):
  File "<input>", line 1, in <module>
  File "/usr/lib/python2.6/dist-packages/numpy/lib/io.py", line 305, in savez
    fid = open(filename,'wb')
IOError: [Errno 2] No such file or directory: '/tmp/data/c.text.npy'

2 个答案:

答案 0 :(得分:2)

您可以在将密钥传递给savez函数之前对其进行编码和解码。

>>> import numpy
>>> import base64
>>> arrs = {'data/a.text': numpy.array([1,2]),
            'data/b.text': numpy.array([3,4]),
            'data/c.text': numpy.array([5,6])}
>>> numpy.savez('file.npz', **dict((base64.urlsafe_b64encode(k), v)
                                    for k,v in arrs.iteritems()))
>>> npzfile = numpy.load('file.npz')
>>> decoded = dict((base64.urlsafe_b64decode(k), v)
                   for k,v in npzfile.iteritems())
>>> decoded
{'data/c.text': array([5, 6]),
 'data/a.text': array([1, 2]),
 'data/b.text': array([3, 4])}

答案 1 :(得分:1)

可能savez使用dict中给出的名称构建临时文件。文件名中包含/。当savez创建文件时,它会尝试使用给定的名称和扩展名.npy(即data/c.txt.py)作为临时目录中的文件名。但是,新路径会导致temp的不存在的子目录,从而导致错误。

解决方案是:用其他东西替换斜杠,或者可以转义文件名。

(我之前的回答太复杂,可能是错误的。)