Go by Example: Sorting by Functions

ときには、コレクションを自然順以外でソートしたいこともあります。 例えば、文字列をアルファベット順ではなく、 長さ順でソートしたいとしましょう。 Go のカスタムソートの例は次の通りです。

 
package main

import (
    "fmt"
    "sort"
)

Go でカスタム関数を使ってソートするためには、 対応する型が必要です。ここでは、byLength 型を作りました。これは、 組み込みの []string 型のただのエイリアスです。

 
type byLength []string

sort パッケージの Sort 関数を使えるように、 sort.Interface すなわち Len, Less, Swap 関数を実装します。 Len と Swap はどの型でもだいたい同じになり、 Less が実際のカスタムソートのロジックをもちます。 この例では、文字列の長さの昇順でソートしたいので、 len(s[i]) と len(s[j]) を使っています。

 
func (s byLength) Len() int {
    return len(s)
}
func (s byLength) Swap(i, j int) {
    s[i], s[j] = s[j], s[i]
}
func (s byLength) Less(i, j int) bool {
    return len(s[i]) < len(s[j])
}

元の fruits スライスを byLength に型変換し、 sort.Sort 関数を使うことでカスタムソートを実現できます。

 
func main() {
    fruits := []string{"peach", "banana", "kiwi"}
    sort.Sort(byLength(fruits))
    fmt.Println(fruits)
}

プログラムを実行すると、期待通り文字列の長さ順にソートされた リストが表示されます。

 
$ go run sorting-by-functions.go
[kiwi peach banana]

ユーザー定義型を作成して 3 つの Interface メソッドを実装し、 sort.Sort 関数を呼び出す、という同じパターンに従うことで、 Go のスライスを任意の関数でソートができます。

Next example: Panic.

by Mark McGranaghan and Eli Bendersky (translated by @oohira) | feedback | license