python - 如何对包含 numpy 数组的元组列表进行排序？

Question

首先，这是不起作用的代码:

ls = [(1.0,np.array([3.0, 4.0])), (1.0,np.array([3.0, 4.1])), (3.0,np.array([2.0, 1.0]))]
ls.sort()

如您所见，我有一个元组列表 (ls)。每个元组的第一个元素是一个 float 。我尝试通过 ls.sort() 对列表进行排序。在大多数情况下，它运作良好。但是，有时(如上例所示)我的元组的第一个元素具有相同的值。在这种情况下，python 尝试使用元组的第二个元素来整理元组，但它不起作用，因为在元组的第二个位置我有 numpy 数组。

如何通过忽略元组的第二个元素来对列表进行排序？如果第一个元素相同，我不关心顺序(可以是原始顺序，也可以是随机的)。

Answer 1

要么告诉python只对第一项排序

sorted(ls, key=lambda t: t[0])

或者将整个东西转换成一个结构化的numpy数组，然后让numpy对其进行排序

ls_arr = np.array(ls, dtype=[('my_val', float), ('my_arr', float, 2)])
ls_arr.sort()

第二个选项仅在数组长度始终相同的情况下才有效。