Python 的 C++ 扩展模块返回一个 Numpy 数组

Question

我正在尝试使用 SWIG 在 C++ 中为 Python 创建一个扩展模块，它可以将 Numpy 数组返回给 Python。 这个 numpy 数组是多维的，每个单元格都可以有一个值或一个值列表（int 64、[int32、int32] 等]。我可以从 C++ 返回一个普通数组到 Python。谁能指导我如何继续这个？我遇到了 numpy.i、 arrayinterface 、typemaps，我不太确定如何开始。有人可以在这方面指导我吗？谢谢。

Answer 1

我认为 Numpy C API 是一个很好的方法。 您仍然可以使用 SWIG。 但我强烈建议不要使用 SWIG 类型映射，它们非常复杂。 相反，您可以在 SWIG 签名中接受并返回 Numpy 数组作为 PyObject* 指针，并使用 C API 构建/提取它们。 SWIG 将正确包装它。

如果这是您的第一个 Python 扩展，请务必阅读Python C API 介绍。 至少，您需要了解用于引用计数的 API 契约。

另一种选择是实现 numpy 理解的缓冲区协议API。

如果您不喜欢 C 代码进入 C++ 世界，那么另一个选择是Boost.Python 。 我没有这方面的经验。 我个人喜欢 Python C API，因为它的复杂性很低。