c++ - pybind11 返回 numpy 对象数组

Question

使用 pybind11 C++ API 和 python3，我们如何在 C++ 实现中正确地创建一个 numpy 对象数组（即 unicode 字符串）并将其返回给 python？传递给 pybind11::array() 的底层数据数组的确切内存布局是什么？我们究竟需要如何管理内存，即删除/释放？

请注意，这是必要的，因为我们希望在创建 pandas DataFrame 时将该字符串数组与其他 POD 数组结合使用。

Answer 1

事实证明，有必要：

1. 创建一个 PyObject 指针数组，填充数组，即

   auto* pbuf = new PyObject*[arraySize]; // or create via pybind11 API...
   pbuf[0] = <new object...>
   pbuf[1] = <new object...>
   etc.
   

2. 创建一个带有胶囊的“对象” py::array() ：

   py::capsule freeWhenDone(pbuf, [](void* pp) {
           delete [] (PyObject*)pp; // or else properly free the pbuf memory
       });
   
   arr = py::array(py::dtype("object"),
       shape, strides, pbuf, freeWhenDone);