python - 使用 numpy.save (和 savez)出现意外类型错误

Question

我正在尝试使用 numpy 的 savez和 load (令人惊讶的是)保存和加载 numpy 数组的方法足够大，以至于我之前使用的方法(保存为 JSON)明显很慢。

我在以下代码片段中重现了我遇到的问题:

import numpy

test_path = "test.npy"
test_data = numpy.random.rand(100000)

with open(test_path, 'w') as test_file:
    numpy.save(test_file, test_data)

运行时，出现以下错误:

TypeError: write() argument must be str, not bytes

我已经能够通过直接将 numpy.save 路径作为字符串传递来解决这个问题:

numpy.save(test_path, test_data)

但是，我发现这个错误很奇怪，因为根据我对文档的阅读，numpy.save 应该直接接受打开的类似文件的对象。我错过了什么吗？

Answer 1

您正在打开文件，然后将其保存到您打开的文件中，numpy.save不需要指向打开文件的文件指针，而是指向文件的路径，因此代码应该是，

numpy.save(test_path, test_data)