如何通过NumPy C API创建简单数组并实现正确内存管理

• 默认情况下，PyArray_SimpleNewFromData 创建的数组不会主动管理传入的内存，也不会在数组被GC回收时自动释放这部分内存，就算你传入的是堆分配内存也一样。
• 直接设置OWNDATA标识的适用条件：只有当你传入的内存是用C标准库malloc/calloc/realloc分配的时，直接开启OWNDATA才是安全的，因为numpy默认释放数据内存时调用的是C的free()，如果你的内存是C++的new[]分配的，直接开OWNDATA会导致free和new[]不匹配，触发未定义行为。
• 针对你用new[]分配内存的场景，正确的处理方式有两种：
  1. 把new[]替换成malloc分配内存，之后调用PyArray_ENABLEFLAGS开启NPY_ARRAY_OWNDATA，numpy就会在数组销毁时自动调用free释放内存。
  2. 如果你必须保留new[]的分配方式，需要为数组绑定自定义析构逻辑：创建一个PyCapsule对象，将你的data指针作为胶囊的内容，胶囊的析构函数中执行delete[] (double*)ptr，再调用PyArray_SetBaseObject把这个胶囊设为数组的基对象，此时数组不需要开OWNDATA，销毁时会自动先释放基对象，触发胶囊的析构函数完成内存释放。

• 你用reinterpret_cast的转换本身语法上是合法的，但是缺少关键的类型校验步骤，存在安全隐患：如果PyArray_SimpleNewFromData调用出错返回了其他类型的对象或者NULL，直接强制转换后访问成员会直接导致程序崩溃。
• 标准的转换流程如下：
  1. 先调用PyArray_Check(obj)校验传入的PyObject*确实是numpy数组类型，校验通过后再执行转换操作。
  2. 校验通过后，你既可以用C风格的强制转换(PyArrayObject*)obj，也可以继续用reinterpret_cast<PyArrayObject*>(obj)，两种写法都符合NumPy C API的使用规范。
• 你补充的代码中已经实现了标志位校验逻辑，只要加上前面的类型检查步骤，就是完全正确的实现。

内容的提问来源于stack exchange，提问作者CiaranWelsh