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

如何解决Python插入/选择排序二维数组

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

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,5,2,3]
]

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

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

如果我的术语有误，请见谅。

解决方法

由于您已经拥有对数组进行排序的函数，因此我将编写一个 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,5,2,3]]
selection2d(arr)
print(arr)
[[1,[2,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]