python

Python二维数组排序的方法有哪些

小亿
186
2024-05-28 15:17:09
栏目: 编程语言

Python中对二维数组进行排序有两种方法:

  1. 使用sorted函数:
# 定义一个二维数组
arr = [[3, 2, 1], [6, 5, 4], [9, 8, 7]]

# 对二维数组按照第一列进行升序排序
sorted_arr = sorted(arr, key=lambda x: x[0])

# 输出排序后的二维数组
for row in sorted_arr:
    print(row)
  1. 使用NumPy库:
import numpy as np

# 定义一个二维数组
arr = np.array([[3, 2, 1], [6, 5, 4], [9, 8, 7]])

# 对二维数组按照第一列进行升序排序
sorted_arr = arr[arr[:,0].argsort()]

# 输出排序后的二维数组
for row in sorted_arr:
    print(row)

这两种方法都可以实现对二维数组的排序,第一种方法使用sorted函数,比较简单直观;第二种方法使用NumPy库,适合处理大规模的数据。

0
看了该问题的人还看了