python中的PyArrow与C++中的Arrow进行接口时如何将Arrow表转换为Arrow表

pip install pyarrow

import pyarrow as pa

# 创建一个示例 Arrow 表
data = {
    'column1': [1, 2, 3],
    'column2': ['a', 'b', 'c']
}
table = pa.Table.from_pydict(data)

# 将 Arrow 表序列化为字节数组
serialized_table = pa.ipc.serialize_table(table).to_buffer()

# 将字节数组保存到文件（或通过其他方式传递给 C++ 代码）
with open('arrow_table.bin', 'wb') as f:
    f.write(serialized_table)

#include <arrow/api.h>
#include <arrow/io/api.h>
#include <arrow/ipc/api.h>
#include <fstream>
#include <iostream>

int main() {
    // 从文件中读取序列化的 Arrow 表
    std::ifstream infile("arrow_table.bin", std::ios::binary);
    std::vector<char> buffer((std::istreambuf_iterator<char>(infile)), std::istreambuf_iterator<char>());

    // 创建 Arrow 内存缓冲区
    auto arrow_buffer = std::make_shared<arrow::Buffer>(reinterpret_cast<const uint8_t*>(buffer.data()), buffer.size());

    // 反序列化 Arrow 表
    std::shared_ptr<arrow::Table> table;
    auto result = arrow::ipc::ReadTable(arrow_buffer);
    if (result.ok()) {
        table = result.ValueOrDie();
    } else {
        std::cerr << "Failed to read Arrow table: " << result.status().ToString() << std::endl;
        return -1;
    }

    // 打印表内容
    std::cout << "Arrow Table:" << std::endl;
    std::cout << table->ToString() << std::endl;

    return 0;
}

cmake_minimum_required(VERSION 3.12)
project(ArrowExample)

find_package(Arrow REQUIRED)

add_executable(ArrowExample main.cpp)
target_link_libraries(ArrowExample Arrow::arrow_shared)

mkdir build
cd build
cmake ..
make
./ArrowExample