如何根据golang中任何给定列中的值对二维数组进行排序？

Question

我们得到一个 NXM 阶的二维数组和一个列号 K ( 1<=K<=m)。 任务是根据 K 列中的值对二维数组进行排序。

 Input : If our 2D array is given as (Order 4X4) 
        39 27 11 42 
        10 93 91 90 
        54 78 56 89 
        24 64 20 65
        Sorting it by values in column 3 
Output : 39 27 11 42 
         24 64 20 65 
         54 78 56 89 
         10 93 91 90 

我认为在这个任务中需要使用 Sort Slice: Len, Less, Swap in Interface 但我不知道具体如何

Answer 1

您实际上可以只使用sort.Slice ，它将逐行对 2D 切片的行进行排序。

sort.Slice采用一个函数来比较每一行。 在此函数中，您可以比较每一行中的列。

https://go.dev/play/p/cj1z7jBs2u5

data := [][]int{
    {39, 27, 11, 42},
    {10, 93, 91, 90},
    {54, 78, 56, 89},
    {24, 64, 20, 65},
}
sortCol := 3
sort.Slice(data, func(i, j int) bool {
    return data[i][sortCol] < data[j][sortCol]
})

---

如果你有一个数组而不是切片，那么你可以使用与上面相同的sort.Slice ，但使用data[:]而不是data 。

https://go.dev/play/p/KcZepTz8SOZ