Python将Numpy数组存为字节文件后，C++无法还原原始数据怎么办？

核心原因 #

你遇到的问题本质是：Python中Numpy数组默认以多字节整数类型（如int32或int64）存储数据，转成字节后每个元素占多个字节；但C++代码仅将文件内容读成单字节的BYTE向量，没有按原始整数类型的字节长度和字节序重新解析数据。

分步解决方案 #

1. 明确Python中Numpy数组的存储类型 #

先在Python代码中添加打印，确认数组的实际数据类型（避免不同平台默认类型差异）：

import numpy as np
f=open("myfile","wb")
mydata=np.array([1,2,3,600])
print(mydata.dtype)  # 输出类似int32或int64
# 更可靠的方式：显式指定类型，避免平台差异
# mydata=np.array([1,2,3,600], dtype=np.int32)
bina = mydata.tobytes()
print(bina)
f.write(bina)
f.close()

2. C++中按对应类型解析字节数据 #

根据Python输出的类型，在C++中将字节数组转换为对应宽度的整数类型（比如int32_t对应np.int32，int64_t对应np.int64），代码示例如下：

#include <fstream>
#include <vector>
#include <cstdint>
#include <iostream>

typedef unsigned char BYTE;
using namespace std;

vector<BYTE> readFile(const char* filename)
{
    streampos fileSize;
    ifstream file(filename, ios::binary);

    file.seekg(0, ios::end);
    fileSize = file.tellg();
    file.seekg(0, ios::beg);

    vector<BYTE> fileData(fileSize);
    file.read((char*)&fileData[0], fileSize);
    return fileData;
}

int main()
{
    vector<BYTE> fileData = readFile("myfile");
    
    // 替换为Python中输出的类型对应的大小：int32_t对应4字节，int64_t对应8字节
    using TargetType = int32_t;
    const size_t elementSize = sizeof(TargetType);
    
    // 校验数据长度合法性
    if (fileData.size() % elementSize != 0) {
        cerr << "数据长度不匹配" << endl;
        return 1;
    }
    
    vector<TargetType> originalData;
    originalData.reserve(fileData.size() / elementSize);
    
    // 按字节序解析（默认小端序，和Numpy默认系统字节序一致）
    for (size_t i = 0; i < fileData.size(); i += elementSize) {
        TargetType num = *reinterpret_cast<const TargetType*>(&fileData[i]);
        originalData.push_back(num);
    }
    
    // 验证结果
    for (int num : originalData) {
        cout << num << " ";
    }
    // 输出应为：1 2 3 600
    
    return 0;
}

关键注意点 #

• 显式指定Numpy数组的dtype：避免不同平台默认类型（如Windows的int32和Linux的int64）导致解析错误。
• 字节序一致性：大多数系统为小端序，Numpy默认使用系统字节序，因此C中直接强制类型转换即可；若需跨平台兼容，可显式指定字节序（如Numpy用mydata.tobytes('big')，C中手动转换大端到小端）。

内容的提问来源于stack exchange，提问作者nivedita