C++读取保存为二进制的 numpy 数组

首先需要确定C++和Python中变量对应的精度类型,
https://docs.scipy.org/doc/numpy/user/basics.types.html#array-types-and-conversions-between-types

常用的,
C++int对应Pythonnp.intc
C++float对应Pythonnp.single
C++double对应Pythonnp.double

numpy数组保存为二进制文件

import numpy as np
a = np.array([[1.888,2.8888,3.88888],[4.666,5.6666,6.66666]])
a = a.astype(np.double)
a.astype('double').tofile("varr.data")

C++读取二进制文件

#include <iostream>
#include <fstream>
using namespace std;

int main()
{
  double fnum[2][3] = {0};

  ifstream in("varr.data", ios::in | ios::binary);
  in.read((char *) &fnum, sizeof fnum);

  // see how many bytes have been read
  cout << in.gcount() << " bytes read\n";

  for(int j=0; j<3; j++) // show values read from file
  cout << fnum[0][j] << " ";

  in.close();

  return 0;
}

如果数据类型对应关系错了,例如把a.astype(‘double‘).tofile("varr.data")改成a.astype(‘float).tofile("varr.data")`,C++中读取的数组将不再是原来的数组。需要特别注意。

相关推荐