用 Python 中的另一个数组对 array 的行进行排序

我尝试用一个数组的值对另一个数组的行进行排序,例如:

import numpy as np
arr1 = np.random.normal(1, 1, 80)
arr2 = np.random.normal(1,1, (80,100))

我希望按降序对 arr1排序,并保持 arr1arr2之间的当前关系(即,在对两者排序之后,arr1[0]arr2[0, :]的行是相同的)。

71352 次浏览

使用 argsort如下:

arr1inds = arr1.argsort()
sorted_arr1 = arr1[arr1inds[::-1]]
sorted_arr2 = arr2[arr1inds[::-1]]

这个例子按降序排序。

使用 zip 函数: zip( *sorted( zip(arr1, arr2) ) )这将做你需要的。

现在的解释是: zip(arr1, arr2)将合并这两个列表,因此得到了[(0,[ ... list 0... ]) ,(1,[ ... list 1... ]) ,... ] 接下来我们运行 sorted(...),默认情况下它根据 tuple 中的第一个字段进行排序。 然后我们再次运行 zip(...),它从已排序的元组中获取元组,并从元组中的第一个元素(来自 arr1)和第二个元素(来自 arr2)创建两个列表。