对两个列表进行排序一个是列表列表

Question

我有两个纽比阵列。一个是N乘M，另一个是N乘1。我希望能够按M维中的任何一个对第一个列表进行排序，并且我希望列表保持相同的顺序（即，如果我交换了list1的第1行和第15行，我希望list2的第1行和第15行也交换）

例如：

import numpy as np
a = np.array([[1,6],[3,4],[2,5]])
b = np.array([[.5],[.8],[.2]])

然后，我希望能够按a中每行的第一个元素进行排序，以给出：

^{pr2}$

或者按a中每一行的第二个元素排序：

^{3}$

我看到很多类似的问题，其中两个列表都是一维的，比如这个question。或者关于列表排序的问题，例如this one。但我找不到我要找的东西。在

最终我得到了这样的结果：

import numpy as np
a = np.array([[1,6],[3,4],[2,5]])
b = np.array([[.5],[.8],[.2]])
package = zip(a,b)
print package[0][1]
sortedpackage= sorted(package, key=lambda dim: dim[0][1])
d,e = zip(*sortedpackage)
print d
print e

现在这就产生了我想要的d和e：

  d = [[3,4],[2,5],[1,6]]
  e = [[.8],[.2],[.5]

但我不明白为什么。print package[0][1]给出了0.5——这不是我排序所依据的元素。为什么会这样？我所做的一切是否稳健？在

Answer 1

在您的package内：

package[0]是(a[0], b[0]) 因此，package[0][1]是b[0]。在

您的包是三重嵌套的。key=lambda dim : dim[0][1]表示使用item[0][1]作为键对package进行排序。package由item组成，item是双嵌套的。在

要查看排序依据的元素，请使用package[x][0][1]x作为该项的索引