argsort
和sort
的区别主要体现在以下几个方面:
- 返回值类型 :
argsort
返回排序后的索引,而sort
返回排序后的值。
- 影响原始张量 :
argsort
不会改变原始张量的值,而sort
会改变原始张量的值。
- 排序维度 :
argsort
可以指定在哪个维度上进行排序,而sort
只能在最后一个维度上进行排序。
- 降序排序 :
argsort
可以通过descending
参数指定是否按降序排序,而sort
可以通过descending=True
参数实现降序排序。
-
argsort
适用于需要获取排序后索引的场景,且不会修改原始数据。 -
sort
适用于需要获取排序后值的场景,会直接修改原始数据。
argsort
;如果不需要保留原始数据并直接获取排序后的值,使用sort
。