Python对按照二维数组第一个元素排序

Python 对二维数组按第一个元素排序是一个常见的需求，特别是在处理数据集、矩阵等场景下。我们可以利用内置的 `sorted()` 函数以及列表推导式来轻松完成这项任务。 ### 使用 sorted() 排序 `sorted()` 是 Python 内置的一个函数，默认会按照每个子列表的第一个元素进行升序排列；如果你想要降序排列，则需要传递额外的关键字参数 `reverse=True` 。此外，我们还可以通过指定关键字参数 `key` 来定制化排序规则，在这个例子中就是基于子列表的第一项进行比较： ```python # 示例：创建一个简单的2D列表 two_d_list = [[4,'d'], [2,'b'], [3,'c'], [1,'a']] # 按照子列表首元素默认从小到大排序 ascending_sorted_2dlst = sorted(two_d_list, key=lambda x: x[0]) print("升序:", ascending_sorted_2dlst) # 如果想从大到小排可以加 reverse 参数 descending_sorted_2dlst = sorted(two_d_list, key=lambda x: x[0], reverse=True) print("降序:", descending_sorted_2dlst) ``` 上面代码将输出: ``` 升序: [[1, 'a'], [2, 'b'], [3, 'c'], [4, 'd']] 降序: [[4, 'd'], [3, 'c'], [2, 'b'], [1, 'a']] ``` 这里使用的 lambda 表达式 `lambda x: x[0]` 定义了如何提取用于比较的关键值——即每个子列表中的首个元素。另外需要注意的是当一维列表里包含非数字类型的元素时（如上述示例），你需要确保这些类型之间能够相互比较，否则可能会抛出异常。

创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考

下一篇 pytorch 进入环境daima