numpy 数组抽取_numpy数组的排序,搜索,元素抽取

排序

numpy的排序函数

sort函数返回排序后的数组

lexsort函数根据键值的字典序进行排序

argsort函数返回数组排序后的下标

ndarray类的sort方法可以对数组进行原地排序

msort函数沿着第一个轴排序

sort_complex函数对复数按照先实部后虚部的顺序进行排序

Key_Function

lexsort函数根据键值的字典序进行排序

将两个数组构成的元组, 分别取出对应下标的元素, 第一个数组的看做key, 第二个数组看做value

然后按照value进行排序, 返回排序后的数组下标

字典序就是按value进行排序

Code

importnumpy as npimportdatetimedefdatestr2num(s):

s= s.decode("utf-8")return datetime.datetime.strptime(s, "%d-%m-%Y").toordinal()

dates, closes= np.loadtxt("AAPL.csv", delimiter=',', usecols=(1, 5),

converters={1:datestr2num}, unpack=True)

indices=np.lexsort((dates, closes))print(indices)'''[ 0 16 1 17 18 4 3 2 5 28 19 21 15 6 29 22 27 20 9 7 25 26 10 8 14

11 23 12 24 13]'''l=[(datetime.date.fromordinal(int(dates[i])), closes[i]) for i inindices]print(l)'''[(datetime.date(2011, 1, 28), 336.10000000000002),

(datetime.date(2011, 2, 22), 338.61000000000001),

(datetime.date(2011, 1, 31), 339.31999999999999),

(datetime.date(2011, 2, 23), 342.62),

(datetime.date(2011, 2, 24), 342.88),

(datetime.date(2011, 2, 3), 343.44),

(datetime.date(2011, 2, 2), 344.31999999999999),

(datetime.date(2011, 2, 1), 345.02999999999997),

(datetime.date(2011, 2, 4), 346.5),

(datetime.date(2011, 3, 10), 346.67000000000002),

(datetime.date(2011, 2, 25), 348.16000000000003),

(datetime.date(2011, 3, 1), 349.31),

(datetime.date(2011, 2, 18), 350.56),

(datetime.date(2011, 2, 7), 351.88),

(datetime.date(2011, 3, 11), 351.99000000000001),

(datetime.date(2011, 3, 2), 352.12),

(datetime.date(2011, 3, 9), 352.47000000000003),

(datetime.date(2011, 2, 28), 353.20999999999998),

(datetime.date(2011, 2, 10), 354.54000000000002),

(datetime.date(2011, 2, 8), 355.19999999999999),

(datetime.date(2011, 3, 7), 355.36000000000001),

(datetime.date(2011, 3, 8), 355.75999999999999),

(datetime.date(2011, 2, 11), 356.85000000000002),

(datetime.date(2011, 2, 9), 358.16000000000003),

(datetime.date(2011, 2, 17), 358.30000000000001),

(datetime.date(2011, 2, 14), 359.18000000000001),

(datetime.date(2011, 3, 3), 359.56),

(datetime.date(2011, 2, 15), 359.89999999999998),

(datetime.date(2011, 3, 4), 360.0),

(datetime.date(2011, 2, 16), 363.13)]'''

对复数进行排序

复数包含实数部分和虚数部分

NumPy中有复数类型, 使用两个浮点数来表示复数.

Key_Function

sort_complex函数对复数进行排序, 按照先实部后虚部的顺序排序

Code

importnumpy as np

np.random.seed(42)#随机数种子, 使用随机数种子, 不管参数如何, 都会生成固定顺序的随机数数组#参数只是起始随机数在随机数数组中的位置#同一个随机数种子, 反复取随机数, 肯定结果是不同的#可能因为所选取的随机数在随机数组中下标在递增

complex_numbers = np.random.random(5) + 1j * np.random.random(5)print(complex_numbers)'''[ 0.37454012+0.15599452j 0.95071431+0.05808361j 0.73199394+0.86617615j

0.59865848+0.60111501j 0.15601864+0.70807258j]'''

print(np.sort_complex(complex_numbers))'''[ 0.15601864+0.70807258j 0.37454012+0.15599452j 0.59865848+0.60111501j

0.73199394+0.86617615j 0.95071431+0.05808361j]'''

numpy数组中搜索

argmax函数返回数组中最大值对应的下标

a = np.array([2, 4, 8])

np.argmax(a)#2

nanargmax函数, 会忽略NaN值

b = np.array([np.nan, 2, 4])

np.nanargmax(b)#2

argmin和nanargmin函数, 返回数组中最小值对应的下标

argwhere函数根据搜索条件搜索非零的元素, 并分组返回对应的下标

a = np.array([2, 4, 8])

np.argwhere(a<= 4)'''[[0]

[1]]'''

searchsort函数可以为指定的插入值寻找维持数组排序的索引位置

extract函数返回满足指定条件的数组元素

Key_Function

searchsorted函数, 返回一个索引位置, 将元素插入这个位置的话, 不改变原数组的有序性

Code

importnumpy as np

a= np.arange(5)print(a)#[0 1 2 3 4]

indices = np.searchsorted(a, [-2, 7])print(indices)#[0 5]

print(np.insert(a, indices, [-2, 7]))#[-2 0 1 2 3 4 7]

numpy中数组元素抽取

Key_Function

extract函数可以根据某个条件从数组中抽取元素.

这个函数和where函数相似

nonzero函数专门用来抽取非零的数组元素

Code

importnumpy as np

a= np.arange(7)print(a)#[0 1 2 3 4 5 6]

condition = (a % 2) ==0print(condition)#[ True False True False True False True]

print(np.extract(condition, a)) #指定了布尔条件从数组中抽取偶数元素#[0 2 4 6]

print(np.nonzero(a))#(array([1, 2, 3, 4, 5, 6]),)