numpy.random.choice - Python

numpy.random.choice是numpy库中的一个随机数生成函数。它可以从一个数组或指定的序列中生成随机样本。

语法

numpy.random.choice(a, size=None, replace=True, p=None)

参数

• a：要从中生成随机样本的序列或数组
• size（可选）：生成的随机样本的大小
• replace（可选）：指定是否采用有/无放回的方式生成随机样本（默认为有放回）
• p（可选）：指定每个元素被选中的概率

返回值

从指定的序列或数组中生成的随机样本。

示例

# 生成一个包含20个元素的一维数组
arr = np.arange(1, 21)

# 生成一个大小为5，有放回的随机样本
sample_with_replacement = np.random.choice(arr, size=5, replace=True)
print(sample_with_replacement)
# [20  5 19 13 13]

# 生成一个大小为5，无放回的随机样本
sample_without_replacement = np.random.choice(arr, size=5, replace=False)
print(sample_without_replacement)
# [16  1 13 15  9]

# 生成一个大小为5，设置每个元素被选中的概率
prob = [0.1, 0.2, 0.3, 0.15, 0.25]
sample_with_probability = np.random.choice(arr, size=5, replace=False, p=prob)
print(sample_with_probability)
# [20  2  9 17 15]

以上示例演示了numpy.random.choice函数的三种不同用法。

• 第一个示例演示了如何生成一个有放回的随机样本。
• 第二个示例演示了如何生成一个无放回的随机样本。
• 第三个示例演示了如何设置每个元素被选中的概率。

总之，numpy.random.choice是一个非常有用的函数，可以快速方便地生成各种类型的随机样本。