Python学习笔记2——Numpy数组的排序和搜索_在numpy中,以下哪个函数可以对数组进行排序

函数说明

sort函数

sort函数能将数组中的数据从小到大进行排序

argsort函数

argsort函数会从小到大返回对应元素的索引值

一维数组

先构建一个一维数组 a（元素随机输入）
用sort函数进行排序，默认升序
需要降序排列的可以用下面方法
利用argsort函数则可以返回数组a中元素从小到大排列的索引值

二维数组

构建一个二维数组 t（元素随机输入）
使用sort函数进行排序，默认会分别将每一行的元素进行升序排序

也可以添加axis参数使其按列进行排序，默认 axis = 1，即按行排序；axis = 0 时，按列排序

拓展

where函数

where 函数类似Excel中的 if 函数，即判断数组中元素是否满足，返回对应值
格式：np.where(条件,x,y)，其中x为条件满足时的输出值，y为条件不满足时的输出值

这里表示在数组 t 中，元素大于5时返回原始值，小于或等于5时返回0

where函数也可用于寻找数组中特定元素的索引值，如下
元素 8 位于数组 t 中第4行第2列

extract函数

extract 函数可用于筛选出满足条件的值

附录代码

import numpy as np
#随机创建数组
a = np.array([4,5,2,8,22,1,6,9,8,4,7,18,12])
a
#将数组 a 进行升序排序
np.sort(a)
#降序排序并将输出的列表转回数组形式
sorted(a,reverse = True)
np.array(sorted(a,reverse = True))
#从小到大返回元素的索引值
np.argsort(a)

#创建二维数组
t = np.array([[3,4,2],[5,9,4],[6,3,2],[4,8,7]])
t
#将数组 t 进行升序排序（按行）
np.sort(t)
#按列
np.sort(t,axis=0)
#按行从小到大返回元素的索引值
np.argsort(t)

#根据条件输出
np.where(t>5,t,0)
#寻找特定元素的索引值
np.where(t==8)
#根据条件筛选元素
np.extract(t>=5,t)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
'
运行
运行