numpy返回索引的函数
时间: 2023-11-15 13:31:42 浏览: 67
numpy中实现ndarray数组返回符合特定条件的索引方法
numpy提供了很多返回索引的函数,下面列举一些常用的:
1. argmin(arr, axis=None, out=None): 返回数组中最小值的索引。
2. argmax(arr, axis=None, out=None): 返回数组中最大值的索引。
3. nonzero(arr): 返回非零元素的索引值。
4. where(condition[, x, y]): 根据条件返回相应元素的索引值。
5. searchsorted(a, v[, side='left', sorter=None]): 在已排序的数组中查找v的插入点,返回索引值。
6. sort(a[, axis, kind, order]): 返回数组排序后的索引值。
7. lexsort(keys[, axis]): 返回多个键的排序后的索引值。
8. argsort(a[, axis, kind, order]): 返回数组排序后的索引值。
9. partition(a, kth[, axis, kind, order]): 返回数组kth小的元素的索引值。
以上是一些常用的返回索引的函数,还有其他的函数可以查阅numpy官方文档。
阅读全文