我正在尝试按另一个数组的值对一个数组的行进行排序。例如:
import numpy as np
arr1 = np.random.normal(1, 1, 80)
arr2 = np.random.normal(1,1, (80,100))
我想按降序对arr1进行排序,并保持arr1和arr2之间的当前关系(即,在对两者进行排序后,arr1[0]
和arr2[0, :]
的行是相同的)
答案 0 :(得分:40)
使用argsort
,如下所示:
arr1inds = arr1.argsort()
sorted_arr1 = arr1[arr1inds[::-1]]
sorted_arr2 = arr2[arr1inds[::-1]]
此示例按降序排序。
答案 1 :(得分:4)
使用zip功能:zip( *sorted( zip(arr1, arr2) ) )
这将满足您的需求。
现在解释:
zip(arr1, arr2)
将两个列表组合在一起,所以你有[(0,[... list 0 ...]),(1,[... list 1 ...]),... ]
接下来我们运行sorted(...)
,默认情况下会根据元组中的第一个字段进行排序。
然后我们再次运行zip(...)
,它从排序中获取元组,并从元组中的第一个元素(来自arr1)和第二个元素(来自arr2)创建两个列表。