Python插入/选择排序二维数组

Question

大家好，我正在开展一个项目，该项目将计算每种排序方法对二维数字数组进行排序的速度。但是我似乎找不到任何可用于对二维数组进行排序的方法。 我的选择排序如下：

def selection(arr): 
    for i in range(len(arr)): 
        min_idx = i 
        for j in range(i+1, len(arr)): 
            if arr[min_idx] > arr[j]: 
                min_idx = j 
                
        arr[i], arr[min_idx] = arr[min_idx], arr[i] 

我的插入代码如下：

def insertionSort(arr): 
    for i in range(1, len(arr)): 
        key = arr[i] 
        j = i-1
        while j >=0 and key < arr[j] : 
                arr[j+1] = arr[j] 
                j -= 1
        arr[j+1] = key

我创建二维数组的方式是：

arrayInput = int(input("please input the number of arrays: \n"))
amountInput = int(input("please enter how many numbers per array: \n"))

arr = np.empty((arrayInput, amountInput))
x = arr
y = x

for i in range(arrayInput):
    for j in range(amountInput):
        arr[i][j] = random.randint(0,100) 

我想知道如何将我的选择和插入排序函数转换为能够对二维数组进行排序。 所以如果我的输入是数组 =

input = [
    [2,4,3,1,5], 
    [3,4,5,2,3]
]

我希望函数通过插入和选择对数组进行排序 输出

expected_output = [
    [1,2,3,4,5], 
    [2,3,3,4,5] 
]

如果我有一些不正确的术语，我们深表歉意。

Answer 1

由于您已经有了对数组进行排序的函数，因此我将编写一个 sort2d 函数，给定其中一个函数，像这样对每一行进行排序：

def sort2d(arr, sort):
   for row in range(len(arr)):
      sort(arr[row])

selection2d = lambda arr: sort2d(arr, selection)
insertionSort2d = lambda arr: sort2d(arr, insertionSort)

arr = [[2,4,3,1,5], [3,4,5,2,3]]
selection2d(arr)
print(arr)
[[1, 2, 3, 4, 5], [2, 3, 3, 4, 5]]

您澄清说您想修改函数本身，因此您使用本地函数内联显示的 sort2d 代码：

 def selection2d(arr);
     def selection(arr): 
         for i in range(len(arr)):
             min_idx = i 
             for j in range(i+1, len(arr)): 
                  if arr[min_idx] > arr[j]: 
                     min_idx = j          
          arr[i], arr[min_idx] = arr[min_idx], arr[i]

      for row in range(len(arr)):
          selection(arr[row])

如果您不喜欢这样，请将 for row in range(len(arr2d)): 添加到选择中，然后将对 arr 的现有引用更改为 arr[row]：

def selection(arr): 
    for row in range(len(arr)):
        for i in range(len(arr[row])): 
            min_idx = i 
            for j in range(i+1, len(arr[row])): 
                if arr[row][min_idx] > arr[row][j]: 
                    min_idx = j 
            arr[row][i], arr[row][min_idx] = arr[row][min_idx], arr[row][i]