我正在嘗試編寫一種算法，該算法將從序列中隨機選擇N個不同的項，而無需事先知道序列的大小，并且在一個以上的序列上進行多次迭代的開銷很大。例如，序列的元素可能是一個巨大文件的行。當N = 1（即“從一個巨大的序列中隨機挑選一個元素”）時，我找到了一種解決方案：import randomitems = range(1, 10) # Imagine this is a huge sequence of unknown lengthcount = 1selected = Nonefor item in items: if random.random() * count < 1: selected = item count += 1但是，對于其他N值（例如N = 3），我該如何實現(xiàn)相同的目標呢？
3 回答
慕無忌1623718
TA貢獻1744條經(jīng)驗 獲得超4個贊
如果您的序列足夠短，以至于可以將其讀入內(nèi)存并對其進行隨機排序，那么一種簡單的方法就是使用random.shuffle：
import random
arr=[1,2,3,4]
# In-place shuffle
random.shuffle(arr)
# Take the first 2 elements of the now randomized array
print arr[0:2]
[1, 3]
根據(jù)序列的類型，您可能需要通過調(diào)用將其轉(zhuǎn)換為列表list(your_sequence)，但是不管序列中對象的類型如何，此方法都可以工作。
自然，如果您無法將序列適合內(nèi)存，或者此方法對內(nèi)存或CPU的要求過高，則需要使用其他解決方案。
茅侃侃
TA貢獻1842條經(jīng)驗 獲得超21個贊
import random
my_list = [1, 2, 3, 4, 5]
num_selections = 2
new_list = random.sample(my_list, num_selections)
# To preserve the order of the list, you could do:
randIndex = random.sample(range(len(my_list)), n_selections)
randIndex.sort()
new_list = [my_list[i] for i in randIndex]
添加回答
舉報
0/150
提交
取消