np.random.choice的用法

最新推荐文章于 2025-05-28 23:37:43 发布

原创最新推荐文章于 2025-05-28 23:37:43 发布 · 220 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#numpy

Python 专栏收录该内容

54 篇文章

订阅专栏

本文深入探讨了NumPy库中random.choice函数的使用方法，包括如何从数组中按一致或非一致分布进行抽样，以及如何控制抽样是否放回。通过具体代码示例，读者可以了解到该函数在数据科学和统计分析中的应用。

import numpy as np

# 参数意思分别 是从a 中以概率P，随机选择3个, p没有指定的时候相当于是一致的分布
a1 = np.random.choice(a=5, size=3, replace=False, p=None)
print(a1)
# 非一致的分布，会以多少的概率提出来
a2 = np.random.choice(a=5, size=3, replace=False, p=[0.2, 0.1, 0.3, 0.4, 0.0])
print(a2)
# replacement 代表的意思是抽样之后还放不放回去，如果是False的话，那么出来的三个数都不一样，如果是

True的话， 有可能会出现重复的，因为前面的抽的放回去了。