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);