在一段代码中,我已经构造了一个numpy数组,并且我想根据列表中指定的特定顺序对该第一个数组进行排序.结果将是第三个数组(new_values).
第一个数组具有要排序的值.
values = numpy.array([10.0, 30.1, 50, 40, 20])
该列表提供了由new_values中的值的索引给出的顺序,该顺序应按降序排列,其中0对应于最大数字.
order=[0, 3, 1, 4, 2].
所以,
new_values[0] > new_values[3] > new_values[1] > new_values[4] > new_values[2]
我试图找到一个特定的排序函数来完成此任务,例如argsort或使用键,但是我不了解如何使它们适应这种情况.
有没有简单的快速方法可以完成此操作,因为我将在许多迭代中执行此操作.如果这有利于使用更好的方法进行排序,我愿意更改第二个数组的顺序,以另一种方式指定索引.
目前,我正在使用以下for循环.
size=len(values)
new_values = np.zeros(size)
values = sorted(values,reverse=True)
for val in range(0, size):
new_values[int(order[val])] = values[val]
先感谢您!
解决方法:
您可以简单地为此使用索引:
>>> import numpy as np
>>> values = np.array([10.0, 30.1, 50, 40, 20])
>>> order=[0, 3, 1, 4, 2]
>>> sorted_array=values[order]
>>> sorted_array
array([ 10. , 40. , 30.1, 20. , 50. ])
同样,如果您需要以下条件,则@Divakar在评论中提到:
new_values[0] > new_values[3] > new_values[1] > new_values[4] > new_values[2]
你可以做 :
>>> values[order]=np.sort(values)[::-1]
>>> values
array([ 50. , 30.1, 10. , 40. , 20. ])
标签:arrays,python,sorting,numpy
来源: https://codeday.me/bug/20191120/2042347.html