numpy 排序，随机数，保留小数

Posted 2023-02-16

参考技术A 1 产生随机数

numpy.random.rand(d0, d1, …, dn)，产生[0,1)之间均匀分布的随机浮点数，其中d0，d1....表示传入的数组形状

2 保留小数

numpy.around( array , decimals=0 , out=None )

decimals等于1的话即为保留一位小数

3排序

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

np.argsort(array, axis=0) #按列排序，如果axis=1则为按行排序

若需要降序排列，可将array改成-array

举例：

import numpy as np

#5*5的二维随机数组

boxes = np.random.rand(5,5)

#保留一位小数

boxes = np.around(boxes,1)

print('boxes=',boxes)

#返回按列排序从大到小的索引值

sql=np.argsort(-boxes,0)

#打印按照第五列从大到小的行索引值

print(sql[:,4])

#打印按照第五列降序排列的新数组

print('boxes after sorting is',boxes[sql[:,4],:])

结果：