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