将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)

但它也不起作用.

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

回溯(最近调用最后一次):file.write(V.shape [0])TypeError:必须是字符串或缓冲区,而不是int

谢谢!

Answer 1

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