numpy数组排序方式

有时候我们需要对numpy中的数据进行排序,选出最大的几个和其所在位置的索引;

1. a.sort()           [a发生变化]

a = np.array([1, 2, 8, 4, 6])
a.sort()
print(a)

[1 2 4 6 8]

2. a1 = numpy.sort(a)             【a不变,a1是排序后的数组】

a = np.array([1, 2, 8, 4, 6])
a1 = np.sort(a)
print(a)
print(a1)

[1 2 8 4 6]
[1 2 4 6 8]

3、numpy.argsort(a, axis=-1, kind='quicksort', order=None)

作用效果:对数组a排序,返回一个排序后索引,a不变

a = np.array([1, 2, 8, 4, 6])
a1 = np.argsort(a)
print(a)
print(a1)

[1 2 8 4 6]
[0 1 3 4 2]

参考:https://blog.csdn.net/panghaomingme/article/details/57933757?utm_source=blogxgwz0

猜你喜欢

转载自blog.csdn.net/weixin_41765699/article/details/83420001
0条评论
添加一条新回复