python - 我怎样才能找到 np.partition() 的实际实现？

Question

了解how numpy.partition() works我正在尝试阅读 source code .

if axis is None:
    # flatten returns (1, N) for np.matrix, so always use the last axis
    a = asanyarray(a).flatten()
    axis = -1
else:
    a = asanyarray(a).copy(order="K")
a.partition(kth, axis=axis, kind=kind, order=order)
return a

它只是用 numpy.asanyarray() 封装输入，并调用自身。

这似乎不是实际的实现，我如何才能找到 np.partition() 的实际实现？

Answer 1

正如您所指出的，np.partition 使用 ndarray 的方法 partition，它本身是用 C 语言编写的，名称为 数组分区。 You can find its source here .如果查看源代码，您会发现它在内部使用了另一个 C 函数，PyArray_Partition，which can be found here .

要查找哪个源文件包含哪个函数，您可以随时使用 GitHub 的搜索栏搜索特定的 repo :-)