浅析python中numpy包中的argsort函数的使用_Python

浅析python中numpy包中的argsort函数的使用

2021-03-30 00:36LJ_Huang Python

这篇文章主要介绍了python中numpy包中的argsort函数的使用，argsort()函数在模块numpy.core.fromnumeric中，本文通过实例代码给大家介绍的非常详细，需要的朋友参考下吧

概述

argsort()函数在模块numpy.core.fromnumeric中。

在python中排序数组，或者获取排序顺序的时候，我们常常使用numpy包的argsort函数来完成。

如下图所示，是使用python获取到数组中的排序的顺序。

									data=numpy.array([1,2,3,4,5])

									datasort=numpy.argsort(data)

									datasort

									Out[39]: array([0, 1, 2, 3, 4], dtype=int64)

									data

									Out[40]: array([1, 2, 3, 4, 5])

									datasort1=data.argsort()

									datasort1

									Out[42]: array([0, 1, 2, 3, 4], dtype=int64)

我们也可以通过help(numpy.argsort)来查看使用方法

									help(numpy.argsort)

									Help on function argsort in module numpy.core.fromnumeric:

									argsort(a, axis=-1, kind='quicksort', order=None)

									  Returns the indices that would sort an array.

									  Perform an indirect sort along the given axis using the algorithm specified

									  by the `kind` keyword. It returns an array of indices of the same shape as

如果想要通过argsort实现排序可以使用切片实现

									data1=numpy.array([1,3,4,56,2,0])

									datasort=data1[data1.argsort()]

									datasort

									Out[48]: array([ 0, 1, 2, 3, 4, 56])

PS：NumPy 中argsort函数

排序函数，返回array类型

argsort函数返回的是数组值从小到大的元素的索引值

									#!/usr/bin/env python

									# -*- coding: utf-8 -*-

									import numpy as np

									inX = np.array([1,2,-1,3,4,7,8])

									print inX

									print inX.argsort()