sort函数返回排序后的数组
lexsort函数根据键值的字典序进行排序
argsort函数返回数组排序后的下标
ndarray类的sort方法可以对数组进行原地排序
msort函数沿着第一个轴排序
sort_complex函数对复数按照先实部后虚部的顺序进行排序
lexsort函数根据键值的字典序进行排序
将两个数组构成的元组, 分别取出对应下标的元素, 第一个数组的看做key, 第二个数组看做value
然后按照value进行排序, 返回排序后的数组下标
字典序就是按value进行排序
import numpy as np import datetime def datestr2num(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 in indices] 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中有复数类型, 使用两个浮点数来表示复数.
sort_complex函数对复数进行排序, 按照先实部后虚部的顺序排序
import numpy 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] ‘‘‘
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函数返回满足指定条件的数组元素
searchsorted函数, 返回一个索引位置, 将元素插入这个位置的话, 不改变原数组的有序性
import numpy 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]
extract函数可以根据某个条件从数组中抽取元素.
这个函数和where函数相似
nonzero函数专门用来抽取非零的数组元素
import numpy as np a = np.arange(7) print(a) # [0 1 2 3 4 5 6] condition = (a % 2) == 0 print(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]),)
原文：https://www.cnblogs.com/draven123/p/11421455.html