如何对包含 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 )。 每个元组的第一个元素是一个浮点数。 我尝试通过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()

第二个选项仅在数组长度始终相同时才有效。

Answer 2

可能使用 key 参数。 这是你想要的吗？

import numpy as np
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(key=lambda x: x[0])