Python二維數(shù)組排序的方法有哪些

小億
186
2024-05-28 15:17:09
欄目: 編程語言

Python中對(duì)二維數(shù)組進(jìn)行排序有兩種方法：

  1. 使用sorted函數(shù)：
# 定義一個(gè)二維數(shù)組
arr = [[3, 2, 1], [6, 5, 4], [9, 8, 7]]

# 對(duì)二維數(shù)組按照第一列進(jìn)行升序排序
sorted_arr = sorted(arr, key=lambda x: x[0])

# 輸出排序后的二維數(shù)組
for row in sorted_arr:
    print(row)
  1. 使用NumPy庫：
import numpy as np

# 定義一個(gè)二維數(shù)組
arr = np.array([[3, 2, 1], [6, 5, 4], [9, 8, 7]])

# 對(duì)二維數(shù)組按照第一列進(jìn)行升序排序
sorted_arr = arr[arr[:,0].argsort()]

# 輸出排序后的二維數(shù)組
for row in sorted_arr:
    print(row)

這兩種方法都可以實(shí)現(xiàn)對(duì)二維數(shù)組的排序，第一種方法使用sorted函數(shù)，比較簡(jiǎn)單直觀；第二種方法使用NumPy庫，適合處理大規(guī)模的數(shù)據(jù)。

0 踩