python生成不重复随机数和对list乱序的解决方法

程序员文章站 2022-03-29 13:56:58

...

下面为大家分享一篇python生成不重复随机数和对list乱序的解决方法，具有很好的参考价值，希望对大家有所帮助。

andom.sample(list, n)即是从list中随机选取n个不同的元素

# -*- coding: utf-8 -*- 
import random 
# 从一个list中随机挑选5个 
list = [12, 23, 13, 14, 78, 234, 123, 12345] 
randomlist = random.sample(list, 5) 
print randomlist 
 
# 在range(10)中随机生成5个不重复的数，可以作为随机下标集合，然后到list中取数 
len = list.__len__() 
indexList = range(len) 
randomIndex = random.sample(indexList, 5) 
for i in randomIndex: 
 print "下标为%d" % i 
 print list[i]

对list进行洗牌，乱序排序，random.shuffle(list)，注意原来的list会被改变

# -*- coding: utf-8 -*- 
import random 
# 对list洗牌，在原list上做改变 
list = range(10) 
print list 
random.shuffle(list) 
print "随机排序列表 : ", list

关于生成随机的不重复数，我被面试过两次，有一次的应用场景是抽奖。

我当时提出的解决方案是：将生成的随机数（数组的随机下标）放在一个数组中，每次对新生成的随机数首先判断是否已经存在，如果不存在就加入这个数组，如果存在就重新生成随机数，直到这个数组中的元素个数到达一个特定值，然后拿着这个随机下标数组到原数组中去取元素。面试官问我时间复杂度问我是多少，我说O(n^2)，面试官问我有没有改进方案，我想了一会没想出来。

回到宿舍后舍友说，你可以每次对选中的元素和最后元素交换以下，下一次生成随机数的时候就在前n-1个元素中生成，这样每次只要交换一次元素，就不用去那个数组中判断当前下标是否已经被生成过一次，时间复杂度变成O(n)，不由佩服。

Java中提供了list.contains(ele)函数，可以直接判断指定容器中是否存在某个元素，这样就不用写二重循环了，但是时间复杂度仍然是O(n^2)

不过今天看了一下python中的sample函数（随机选取种子点）直接就能达到我要的结果，下次再写抽奖函数一行代码就搞定了。