登录
推荐 文章 Go 技术 课程 下载 专题 AI
首页 >  Golang >  Go教程

Go 怎么随机打乱切片并抽取不重复的元素

来源:17golang原创

时间:2026-09-06 07:26:35 106浏览 收藏

在 Go 里,这个问题最好拆成两个动作:如果要改变已有切片的排列,用 math/rand/v2rand.Shuffle;如果要抽取 k 个不重复的位置,用 rand.Perm 先生成随机下标,再读取前 k 项。后者不会因为“随机数碰巧重复”而陷入重试,也能清楚处理数量边界。

普通抽样优先使用 ShufflePerm;需要保留原切片时先复制,涉及令牌、验证码等不可预测结果时改用 crypto/rand

本文示例使用 math/rand/v2。它产生的是伪随机数,适合模拟、抽样和普通业务随机化,不应承担安全随机数职责。

切片随机化和抽样其实是两个动作

“随机打乱”要求所有元素仍然保留,只是顺序变化;“随机抽取”只需要得到部分元素,而且每个位置最多取一次。两者都可以建立在随机排列上,但副作用不同:

目标推荐 API是否修改原切片结果特点
打乱全部元素rand.Shuffle长度不变,元素仍是原集合
抽取不重复元素rand.Perm先得到不重复下标,再取前 k 项
原始切片、rand.Shuffle 和交换回调之间的静态关系
图1:语义色块关系图展示原始切片、rand.Shuffle、swap 回调和新排列之间的职责关系;它是结构说明图,不是运行截图。

用 rand.Shuffle 原地打乱已有切片

Shuffle 接收元素数量和一个交换函数。回调中的两个下标由随机化过程传入,业务代码只负责交换对应元素:

package main

import (
	"fmt"
	"math/rand/v2"
)

func main() {
	items := []string{"北京", "上海", "广州", "深圳"}

	// Shuffle 会直接修改 items 的元素顺序,不会创建新的切片。
	rand.Shuffle(len(items), func(i, j int) {
		items[i], items[j] = items[j], items[i]
	})

	fmt.Println(items)
}

这里的关键是“原地”:如果其他代码还持有 items 的引用,它看到的顺序也会改变。切片为空时 len(items) 为 0,调用仍然安全;但不要在回调里访问固定下标或修改切片长度。

若需要保留原顺序,可以先复制,再对副本洗牌:

shuffled := append([]string(nil), items...)

// 只打乱副本,items 仍保持原来的顺序。
rand.Shuffle(len(shuffled), func(i, j int) {
	shuffled[i], shuffled[j] = shuffled[j], shuffled[i]
})

用 rand.Perm 抽取不重复下标

抽样不必循环调用 IntN 再用 map 去重。rand.Perm(n) 直接返回 0n-1 的随机排列,因此取前 k 个下标就能保证“按位置不重复”:

package main

import (
	"fmt"
	"math/rand/v2"
)

func Pick(items []string, k int) ([]string, bool) {
	// k 不在合法区间时返回 false,避免切片越界。
	if k  len(items) {
		return nil, false
	}

	indexes := rand.Perm(len(items))
	selected := make([]string, 0, k)
	for _, index := range indexes[:k] {
		// 每个 index 在排列中只出现一次,因此位置不会重复。
		selected = append(selected, items[index])
	}
	return selected, true
}

func main() {
	items := []string{"A", "B", "C", "D", "E"}
	selected, ok := Pick(items, 3)
	if !ok {
		return
	}
	fmt.Println(selected)
}

k == 0 会得到空结果,k == len(items) 会得到全部元素的随机排列。需要注意的是,“不重复”针对的是下标;如果原切片里有两个相同字符串,它们位于不同位置,结果仍可能出现相同的值。

候选切片、rand.Perm、不重复下标和抽样结果之间的静态关系
图2:语义色块关系图把候选切片、随机排列、抽样数量和结果集合分开,帮助判断不重复保证来自下标排列,而非字符串值去重。

什么时候应该换成 crypto/rand

math/rand/v2 的顶层函数适合普通并发调用,但它仍是伪随机源,不能用来生成密码重置令牌、验证码、会话标识或需要抗预测的抽签结果。此类场景应使用 crypto/rand,并按其 API 读取随机字节或随机整数。

还要留意两个边界:第一,rand.Perm 会创建与候选数量相关的下标切片,超大集合应评估内存;第二,多 goroutine 共享自建的 RandSource 需要同步,顶层函数和自建随机器的并发规则不要混用。

常见问题

洗牌后还能拿到原切片吗?能,但它已经被原地改序;想同时保留两个版本就先复制。

为什么不用循环 IntN 直到拿够 k 个?那种写法需要额外的去重集合,候选接近取完时还会不断碰撞;Perm 把唯一性放在下标排列里,边界更直接。

速记:改顺序用 Shuffle,抽不重复位置用 Perm,保留原数据先复制,安全场景使用 crypto/rand

声明:本文转载于:17golang原创 如有侵犯,请联系study_golang@163.com删除
相关阅读
更多>
最新阅读
更多>
课程推荐
更多>