np.random.choice()是一个在NumPy库中用于生成随机样本的函数。它可以从一个给定的1-D数组中生成多个随机数。
例如,你可以使用它来从一个数字列表中生成一个随机数:
import numpy asnp
# 生成一个0到9的随机数
num = np.random.choice(10)
print(num)
你也可以设置概率分布,让某些数字更可能被选中:
import numpy as np
# 从0到9中生成一个随机数,让5更可能被选中
num = np.random.choice(10, p=[0.1, 0.1, 0.1, 0.1, 0.1, 0.1, 0.1, 0.1, 0.1, 0.2])
print(num)
你还可以设置返回的样本数量,生成多个随机数:
import numpy as np# 从0到9中生成三个随机数
nums = np.random.choice(10, size=3)
print(nums)
有关np.random.choice()函数的更多信息,可以参考NumPy官方文档:https://numpy.org/doc/stable/reference/random/generated/numpy.random.choice.html