根据权重,从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]