从字典的每个键中随机采样一个元素

Question

我有一个字典，其中键代表组，值是元素列表（为简单起见，我们假设值是整数）。

这样的字典的一个例子，d，是：

d = {2: [0, 1, 7, 8, 9], 1: [2, 4], 4: [3], 3: [5, 6]}

我希望对这本字典进行多次随机抽样，其中样本将包括对第一组中的单个元素、第二组中的单个元素等的随机抽样。

简单的输出示例是：

[9, 2, 3, 5]
[1, 2, 3, 6]
[7, 4, 3, 6]
...

我可以遍历字典，d，M 次并遍历键和样本 1 元素，我想知道是否有更简单的方法，希望更有效，因为它将成为更大更复杂的算法。

天真的方法是：

ll = []
    for i in range(10):
        tmp = []
        for k, l in d.items():
            tmp.append(random.choice(l))
        ll.append(tmp)
    print(ll)  # [[5, 6, 9, 7], [8, 4, 3, 7], [2, 1, 3, 7], [5, 1, 3, 7], [0, 4, 3, 7], [2, 1, 3, 7], [2, 6, 9, 7], [2, 1, 9, 7], [8, 4, 9, 7], [2, 1, 9, 7]]

我对字典并不严格，我也可以使用其他结构，但逻辑保持不变。

不胜感激。

Answer 1

实际上没有其他选择，因为您正在使用字典和列表。正如您所提到的，您必须遍历字典值M 次，并在每次迭代中获取每个内部列表的random.choice。一个简单的方法是：

from random import choice
n = 5
l = list(d.values())
[list(map(random.choice, l)) for _ in range(n)]
# [[9, 2, 3, 6], [7, 2, 3, 5], [9, 4, 3, 5], [0, 2, 3, 5], [7, 4, 3, 5]]

另一种方法是使用random.choices 从内部列表中进行替换，然后使用zip转置生成的嵌套列表。这样，我们将上面的内容简化为对值的单次迭代：

from random import choices
list(zip(*(random.choices(i, k=n) for i in l)))
# [(8, 4, 3, 5), (1, 2, 3, 6), (9, 4, 3, 6), (1, 2, 3, 5), (8, 4, 3, 5)]

Answer 2

这应该可行：

import random

result=[]
for i in range(M):
    result.append([random.choice(d[i]) for i in d])