numpy.argsort()实现位置查找与疑似循环引用问题咨询

1. 是否存在循环引用？ #

你追踪到的调用链路不属于实际运行中的循环引用。numpy/array_api目录下的代码是numpy为了兼容Array API通用标准做的适配层，只有你显式使用numpy的array API命名空间实例时才会调用这部分代码，普通numpy.ndarray调用argsort时完全不会走到这个路径，因此不存在实际生效的循环调用。

2. 第二步的猜测是否错误？ #

这个猜测是错误的。array_api.__init__.py下的argsort是适配层接口，和原生numpy.ndarray的内置方法无关，原生ndarray的方法不会在这个目录下实现。

3. numpy.argsort的实际实现位置 #

• Python层入口：确实如官方文档标注，在numpy.core.fromnumeric.py中，调用逻辑是numpy.argsort(arr) -> 调用_wrapfunc -> 转发到arr.argsort()实例方法。
• 底层实现：ndarray.argsort是C语言实现的方法，核心排序逻辑位于numpy源码的numpy/core/src/npysort/目录下，包含了快速排序、归并排序、堆排序、Timsort等多种排序算法的C实现，根据你调用时指定的kind参数选择对应的排序逻辑生成索引数组。

如果你的输入数组已经完成预排序，完全不需要调用np.unique走默认逻辑，可以直接用np.diff配合边界判断提取唯一值和对应索引，完全跳过argsort步骤，性能会有数量级提升。

内容的提问来源于stack exchange，提问作者Azriel 1rf