Go 通过结构体指定字段进行排序

对结构体指定字段进行排序:

  对结构体指定字段进行排序: 

package main

import (
	"fmt"
    "sort"
)

// 对结构体指定字段进行排序
type User struct {
	Name string `json:"name"` // `json:"xxx"`:在结构体和json字符串字段顺序不一致的情况下:unmarshal根据tag去寻找对应字段的内容
	Age  int    `json:"age"`
}

// type Users []User
// func SortByAge(u Users) {
func SortByAge(u []User) {
	fmt.Printf("源数据:%+v\n", u)

	sort.Slice(u, func(i, j int) bool { // desc
		return u[i].Age > u[j].Age
	})
	fmt.Printf("按Age降序:%+v\n", u)

	sort.Slice(u, func(i, j int) bool { // asc
		return u[i].Age < u[j].Age
	})
	fmt.Printf("按Age升序:%+v\n", u)
}

func main() {
	// 初始化结构体对象数组:
	// 初始化方法1:
	// users := Users{
	// 	{
	// 		Name: "test1",
	// 		Age:  22,
	// 	},
	// 	{
	// 		Name: "test2",
	// 		Age:  19,
	// 	},
	// 	{
	// 		Name: "test3",
	// 		Age:  25,
	// 	},
	// }

	// 初始化方法2:
	var users []User
	var u User
	u.Name = "test1"
	u.Age = 22
	users = append(users, u)
	u.Name = "test2"
	u.Age = 20
	users = append(users, u)
	u.Name = "test3"
	u.Age = 26
	users = append(users, u)

	SortByAge(users)
}

// 输出:
源数据:[{Name:test1 Age:22} {Name:test2 Age:20} {Name:test3 Age:26}]
按Age降序:[{Name:test3 Age:26} {Name:test1 Age:22} {Name:test2 Age:20}]
按Age升序:[{Name:test2 Age:20} {Name:test1 Age:22} {Name:test3 Age:26}]

此外也可使用sort.Sort()方法,不过需要自己去实现 Len()、Swap()、Less()方法,参考:golang对自定义类型排序

另外,通过借助“结构体指定字段进行排序”解了一道LeetCode 347题:传送门

补充示例:同时对多个字段进行排序(升序/降序)

package main

import (
	"fmt"
	"sort"
)

type User struct {
	Name  string
	High  int
	Wight int
}

// 先根据身高 High 排序,在High相等时,再根据体重 Wight 排序,最后返回排序后结果
func TestSort(userArr []User) []User {
	for i, val := range userArr {
		userArr[i] = User{Name: val.Name, High: val.High, Wight: val.Wight}
	}

	sort.Slice(userArr, func(i, j int) bool { // 根据High降序
		return userArr[i].High > userArr[j].High
	})

	sort.Slice(userArr, func(i, j int) bool {
		if userArr[i].High == userArr[j].High { // 当High相等时
			return userArr[i].Wight > userArr[j].Wight // 根据Wight降序
		} else {
			return userArr[i].High > userArr[j].High
		}
	})

	return userArr
}

func main() {
	userInfoArr := TestSort([]User{
		{
			Name:  "user1",
			High:  170,
			Wight: 60,
		},
		{
			Name:  "user2",
			High:  170,
			Wight: 80,
		},
		{
			Name:  "user1",
			High:  185,
			Wight: 60,
		},
	})
	fmt.Println("userInfoArr: ", userInfoArr)
}

评论 13
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值