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

Question

我正在尝试按另一个数组的值对一个数组的行进行排序.例如:

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, :]它们是相同的).

Answer 1

使用argsort方法如下:

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

它按降序排序.

Answer 2

使用zip功能：zip( *sorted( zip(arr1, arr2) ) ) 这将满足您的需求。

现在的说明： zip(arr1, arr2)将两个列表合并，所以您有[（0，[... list 0 ...]），（1，[... list 1 ...]），...]接下来，我们运行sorted(...)，默认情况下，它基于元组的第一个字段进行排序。然后，我们zip(...)再次运行，从中对元组进行排序，并从元组中的第一个元素（来自arr1）和第二个元素（来自arr2）创建两个列表。