Python中numpy function对应的C-api名称是什么

Question

• 我想为Python写一个C扩展库，旨在用C替换Python代码。
• Python 代码有几行如下所示：

import numpy as np
a = np.array([1,3,12,0.43,234,-3,-4])
b = a[[1,3,5]]
print(b)

# array([ 3.  ,  0.43, -3.  ])

• 与将 int 作为 numpy 数组的索引不同，此示例将数组视为索引。

• 我很困惑获取给定 numpy 数组的指定索引，对应的 C-API 名称是什么？

• NUMPY C-API 文件在numpy c-api

• 非常感谢。

Answer 1

高级 Numpy 函数并不意味着从 C 开始使用。事实上，并非所有这些函数都在 C 中实现（一些是在纯 Python 中调用其他高级 function 本身在 C 中实现）。 事实上，除了重用一些与直接在 C 中实现的代码相比效率不是很高的代码，这样做并没有太大的好处。

Numpy 提供了一个非常小的接口（您提供的接口），因此模块可以在 Numpy 阵列上运行。 在实践中，低级 C 模块通常使用PyArray_GetPtr （或更具体的宏，如PyArray_GETPTR2 ，同时检查标志以确保安全）提取指针，并根据步幅直接对数组缓冲区进行操作（使用PyArray_STRIDES提取）。 Numpy 数组只是一个具有固定大小和一些元信息的大型原始缓冲区。 视图添加了更多信息，如维数、形状、步幅等。如果可能，最好检查数组是否连续，并为连续的 arrays 编写代码。确实，编译器往往不会在某些情况下生成快速代码步幅设置为 1：您有责任在 C 中对其进行优化（这是 Numpy 在其 function 中所做的，但这部分引入了一些开销）。 针对连续数组进行优化的代码可以更快（主要是由于可能使用 SIMD 指令和更快的索引指令）。

虽然您可以使用PyArray_GETITEM ，但由于可能的检查、低效的通用索引（破坏许多编译器优化）以及 function 肯定不会被编译器内联，生成的代码将比直接访问慢得多。 它可能比纯 Python 代码更快，但幅度不大（当然类似于不使用直接索引的 Cython 代码）。

Answer 2

• 最后我找到了这个 numpy C 函数，它的名字是PyArray_Choose 。
• 完整的函数定义是PyObject *PyArray_Choose(PyArrayObject *self, PyObject *op, PyArrayObject *ret, NPY_CLIPMODE clipmode)
• numpy 选择