Python insertion/selection 对二维数组进行排序

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]

Python insertion/selection 对二维数组进行排序

Python insertion/selection sort 2d arrays

python

arrays

multidimensional-array

insertion-sort

selection-sort