将numpy数组转换为c++本机向量

#include <iostream>
#include <vector>
#include <numpy/arrayobject.h>

std::vector<double> numpyToVector(PyObject* numpyArray) {
    PyArrayObject* array = reinterpret_cast<PyArrayObject*>(numpyArray);
    int size = PyArray_SIZE(array);
    double* data = static_cast<double*>(PyArray_DATA(array));

    std::vector<double> vector(data, data + size);
    return vector;
}

#include <iostream>
#include <vector>
#include <Python.h>
#include <numpy/arrayobject.h>

std::vector<double> numpyToVector(PyObject* numpyArray) {
    // 与前面的代码相同
}

int main() {
    // 初始化Python解释器
    Py_Initialize();

    // 导入numpy模块
    import_array();

    // 创建一个numpy数组
    PyObject* numpyArray = PyArray_SimpleNew(1, new npy_intp[1]{5}, NPY_DOUBLE);
    double* data = static_cast<double*>(PyArray_DATA(reinterpret_cast<PyArrayObject*>(numpyArray)));
    for (int i = 0; i < 5; i++) {
        data[i] = i + 1;
    }

    // 将numpy数组转换为C++本机向量
    std::vector<double> vector = numpyToVector(numpyArray);

    // 打印向量的内容
    for (double value : vector) {
        std::cout << value << " ";
    }
    std::cout << std::endl;

    // 释放numpy数组
    Py_DECREF(numpyArray);

    // 清理Python解释器
    Py_Finalize();

    return 0;
}