Python杂谈 | (9) np.random.choice()函数用法

目录

 

1. 函数简介

2. 使用例子

---

1. 函数简介

numpy中的np.random.choice(arr,size=None,replace=True,p=None),通过给定的一维数组数据产生随机采样。

参数：

• arr: 一维数组或int型变量。若是数组就按照数组中的数据进行采样，若是int型变量，相当于np.arange(a)。
• size：int型变量或者 tuple of ints 决定输出的shape。如果给定的是(m,n,k),那么m*n*k个采样点将会被采样。
• replace：采样是否放回，即决定采样中是否有重复值
• p：一维数组，对应arr数组中每个采样点的概率分布，即每个采样点以多大的概率被采样。没有给出则使用标准分布，每个采样点等概率被采样。

返回值：

samples：单一数值或ndarray

 

2. 使用例子

import numpy as np

#从0-4中等概率采样3个数 可以重复 
print(np.random.choice(5,3))
#从0-4中等概率采样3个数 不可以重复
pr