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

2023年7月30日 227次阅读

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

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必须是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]
点赞