numpy：np.random.choice的用法

最新推荐文章于 2025-10-17 13:37:57 发布

原创最新推荐文章于 2025-10-17 13:37:57 发布 · 5.3k 阅读

9 ·

CC 4.0 BY-SA版权

文章标签：

#numpy #np.random.choice

人工智能专栏收录该内容

62 篇文章

订阅专栏

本文介绍了如何使用Python中的numpy库进行随机抽样操作。详细解释了numpy.random.choice函数的用法，包括设置不同的参数来控制抽样过程，如设置是否放回抽样、指定抽样的分布概率等。

import numpy as np

# 参数意思分别 是从a 中以概率P，随机选择3个,从0开始选，取不到a，最多取到a-1. p没有指定的时候相当于是一致的分布
a = np.random.choice(a=5, size=3, replace=False, p=None)
print(a)
# 非一致的分布，会以多少的概率提出来，出现0点概率为0.2，出现1点概率为0.1，以此类推，出现4点概率为0.0，没有5，只有0-4.
b= np.random.choice(a=5, size=3, replace=False, p=[0.2, 0.1, 0.3, 0.4, 0.0])
print(b)
# replacement 代表的意思是抽样之后还放不放回去，如果是False的话，那么出来的三个数都不一样，如果是
#True的话， 有可能会出现重复的，因为前面的抽的放回去了。