numpy.argwhere文档中“按元素分组”含义及文档准确性问询

你遇到的差异不是函数特性遗漏，也不是文档表述错误，而是对文档中「按元素分组」的含义存在理解偏差。

• 首先明确argwhere的底层逻辑：函数实现return transpose(nonzero(a))完全符合设计预期。np.nonzero会按**行优先（C顺序）**遍历数组，返回所有非零元素各维度的独立索引元组，转置操作只是把每个元素对应的多维度索引打包成单独的子数组，不会改变元素遍历的先后顺序。你测试用例中数组所有元素都是非零值，所以输出顺序就是按行遍历的顺序，和你实际得到的结果一致。
• 文档中「按元素分组」的正确含义：这里的分组不是按数组元素的值进行分组，而是指将同一个非零元素对应的多维度索引打包为一组。以你的测试用例为例，np.nonzero(x)返回的是(array([0,0,0,1,1,1]), array([0,1,2,0,1,2]))，两组独立的行/列索引，经过转置后每个位置的行、列索引被合并为[行号,列号]的子数组，这就是文档描述的「按元素分组」。
• 如果你需要按元素值分组的索引结果，需要额外自行实现逻辑，argwhere本身不提供按值排序或分组的功能。

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