如何從列表中返回一個或多個項目？

Question

目前我的代碼返回客戶端的名稱和visit_length的隨機值。我想它返回客戶名稱，然後從visit_length一個或多個元素，即("Client 1", 15, 45), ("Client 2", 45, 60), ("Client 3", 30)

N = 20  
randomitems = [] 
visit_length = [15, 30, 45, 60] 
value_range = np.arange(0, N, 1) 

for i in value_range: 
    visits = ("Client %d" % i, random.choice(visit_length)) 
    randomitems.append(visits) 

有什麼建議？我想隨機庫裏可能有些東西，但我還沒有找到任何東西。

謝謝！

Answer 1

選擇一個隨機數k的項目追加random.randint，然後從visit_length與random.sample中獲取該元素的數量。

import random 
import numpy as np 

N = 20 
randomitems = [] 
visit_length = [15, 30, 45, 60] 
value_range = np.arange(0, N, 1) 

for i in value_range: 
    k = random.randint(1, len(visit_length)) 
    visits = ("Client %d" % i,) + tuple(sorted(random.sample(visit_length, k=k))) 
    randomitems.append(visits) 

如果你想允許重複項目：

import random 
import numpy as np 

N = 20 
MAX_VISIT_LENGTHS = 4 # You can choose as many as you want now I guess 
randomitems = [] 
visit_length = [15, 30, 45, 60] 
value_range = np.arange(0, N, 1) 

for i in value_range: 
    k = random.randint(1, MAX_VISIT_LENGTHS) 
    vs = sorted(random.choice(visit_length) for _ in range(k)) 
    # If you do not want to sort use 
    # vs = [random.choice(visit_length) for _ in range(k)] 
    visits = ("Client %d" % i,) + tuple(vs) 
    randomitems.append(visits) 

Answer 2

更新答案返回元組

我可能誤會了你想要的，但這裏是我對這個問題： 我們創建一個函數，它從列表x中返回隨機數量的項目（在1-4 = N之間），並以隨機順序返回它們。

import random 

def visits_func(): 
    N = random.randint(1, 4) 
    x = [15, 30, 45, 60] 
    y = [] 
    while N: 
     y.append(x.pop(random.randrange(len(x)))) 
     N -= 1 
    return y 

value_range = range(1,6) #1,2,3,4,5,6 

randomitems = [] 

for i in value_range: 
    visits = tuple(["Client %d" % i] + sorted(visits_func())) 
    randomitems.append(visits) 

打印，例如，

[('Client 1', 15, 60), 
('Client 2', 15), 
('Client 3', 45, 60), 
('Client 4', 30, 60), 
('Client 5', 15, 30, 45)]