python - 将 numpy 数组及其大小写入二进制文件

Question

我需要将一个 2D numpy 数组写入一个文件，包括它的维度，这样我就可以从 C++ 程序中读取它并创建相应的数组。

我写了一些简单的代码来保存数组，它可以从 C++ 中读取，但是如果我尝试先写数组的大小，它总是给我一个错误。

这是我的简单 python 代码:

1 file = open("V.bin","wb")
2 file.write(V.shape)
3 file.write(V)
4 file.close()

第二行报错，我也试过了:

n1, n2 = V.shape
file.write(n1)
file.write(n2)

但它也不起作用。

我正在添加它显示的错误:

追溯(最近的调用最后): 文件.write(V.shape[0]) TypeError: 必须是 string 或 buffer，不是 int

谢谢!

Answer 1

您可以使用 numpy.save()以二进制形式保存。