numpy.random.choice采樣（可按數據的概率分布采樣）

函數解析

numpy.random.choice(a,  # 從a中采樣，必須是一維或者是int
                     size=None,   # 采樣數據的形狀，可以是tuple或int
                     replace=True,   # True表示有放回抽樣， False表示無放回
                     p=None)  # 表示a中每個元素被抽樣的概率，和a的維度要一樣，默認每個元素的概率相等

舉例

# 從range(5)的范圍中隨機取3個數字
>>> np.random.choice(5, 3)
array([3, 0, 4])

# 從x中以概率分布為prob，有放回地采樣2個數字
>>> x = np.array([1, 2, 3])
>>> prob = np.array([0.8, 0.1, 0.1])
>>> np.random.choice(a=x, size=2, replace=True, p=prob)
array([2, 1])