根据权重，从python中的列表列表中随机选择列表

Question

我有一个列表列表，其中每个列表都与分数/重量相关联。我想通过从第一个列表中随机选择来生成一个新的列表列表，以便那些得分较高的列表更频繁地出现。当population只是一个普通列表时，下面的行正常工作。但我希望将它列入清单。

population = [['a','b'],['b','a'],['c','b']]
list_of_prob = [0.2, 0.2, 0.6]

population = np.random.choice(population, 10, replace=True, p=list_of_prob)

这将给出输出ValueError: a must be 1-dimensional

Answer 1

不是传递实际列表，而是将带有索引的列表传递到列表中。

np.random.choice已经允许这样做，如果你传递一个int n，那么就像你通过np.arange(n)一样。

所以

choice_indices = np.random.choice(len(population), 10, replace=True, p=list_of_prob)
choices = [population[i] for i in choice_indices]