python - 如何在元组列表中使用 numpy.random.choice？

Question

我需要以给定的概率随机选择列表中的元组。

编辑:每个元组的概率在概率列表中不知道忘了参数replacement，默认是none使用数组而不是列表的相同问题

下一个示例代码给我一个错误:

import numpy as np

probabilit = [0.333, 0.333, 0.333]
lista_elegir = [(3, 3), (3, 4), (3, 5)]

np.random.choice(lista_elegir, 1, probabilit)

错误是:

ValueError: a must be 1-dimensional

我该如何解决？

Answer 1

根据函数的文档，

a : 1-D array-like or int
    If an ndarray, a random sample is generated from its elements.
    If an int, the random sample is generated as if a was np.arange(n)

接下来

lista_elegir[np.random.choice(len(lista_elegir),1,p=probabilit)]

应该做你想做的。 (p= 根据注释添加；如果值统一则可以省略)。

它从 [0,1,2] 中选择一个数字，然后从您的列表中选择该元素。