基于 Python 的 8 种常用抽样方法

小敢摘葡萄

已于 2024-01-23 21:42:28 修改

阅读量3.2k

点赞数

文章标签： python 机器学习开发语言抽样方法 Python编程 Python学习 python入门

于 2023-11-04 11:40:42 首次发布

本文链接：https://blog.csdn.net/xiaoganbuaiuk/article/details/134216249

版权

本文介绍了统计学和机器学习中的抽样方法，包括概率抽样（随机抽样、分层抽样、聚类抽样、系统抽样）和非概率抽样（简单采样、自愿抽样、雪球抽样），强调了在实际应用中概率抽样在数据分析中的重要性，特别是在信贷风险控制中的样本设计。

摘要由CSDN通过智能技术生成

抽样是统计学、机器学习中非常重要，也是经常用到的方法，因为大多时候使用全量数据是不现实的，或者根本无法取到。所以我们需要抽样，比如在推断性统计中，我们会经常通过采样的样本数据来推断估计总体的样本。

上面所说的都是以概率为基础的，实际上还有一类非概率的抽样方法，因此总体上归纳为两大种类：

概率抽样：根据概率理论选择样本，每个样本有相同的概率被选中。
非概率抽样：根据非随机的标准选择样本，并不是每个样本都有机会被选中。

概率抽样技术

1.随机抽样(Random Sampling)

这也是最简单暴力的一种抽样了，就是直接随机抽取，不考虑任何因素，完全看概率。并且在随机抽样下，总体中的每条样本被选中的概率相等。

比如，现有10000条样本，且各自有序号对应的，假如抽样数量为1000，那我就直接从1-10000的数字中随机抽取1000个，被选中序号所对应的样本就被选出来了。

在Python中，我们可以用random函数随机生成数字。下面就是从100个人中随机选出5个。

import random  
population = 100  
data = range(population)  
print(random.sample(data,5))  
> 4, 19, 82, 45, 41

2.分层抽样(Stratified Sampling)

分层抽样其实也是随机抽取，不过要加上一个前提条件了。在分层抽样下，会根据一些共同属性将带抽样样本分组，然后从这些分组中单独再随机抽样。

因此，可以说分层抽样是更精细化的随机抽样，它要保持与总体群体中相同的比例。 比如，机器学习分类标签中的类标签0和1，比例为3:7，为保持原有比例，那就可以分层抽样，按照每个分组单独随机抽样。

Python中我们通过train_test_split设置stratify参数即可完成分层操作。

from sklearn.model_selection import train_test_split  
  
stratified_sample, _ = train_test_split(population, test_size=0.9, stratify=population[['label']])  
print (stratified_sample)

3.聚类抽样(Cluster Sampling)

聚类抽样，也叫整群抽样。它的意思是，先将整个总体划分为多个子群体，这些子群体中的每一个都具有与总体相似的特征。也就是说它不对个体进行抽样，而是随机选择整个子群体。

用Python可以先给聚类的群体分配聚类ID，然后随机抽取两个子群体，再找到相对应的样本值即可，如下。

import numpy as np  
clusters=5  
pop_size = 100  
sample_clusters=2  
# 间隔为 20, 从 1 到 5 依次分配集群100个样本的聚类 ID，这一步已经假设聚类完成  
cluster_ids = np.repeat([range(1,clusters+1)], pop_size/clusters)  
# 随机选出两个聚类的 ID  
cluster_to_select = random.sample(set(cluster_ids), sample_clusters)  
# 提取聚类 ID 对应的样本  
indexes = [i for i, x in enumerate(cluster_ids) if x in cluster_to_select]  
# 提取样本序号对应的样本值  
cluster_associated_elements = [el for idx, el in enumerate(range(1, 101)) if idx in indexes]  
print (cluster_associated_elements)

4.系统抽样(Systematic Sampling)

系统抽样是以预定的规则间隔（基本上是固定的和周期性的间隔）从总体中抽样。比如，每 9 个元素抽取一下。一般来说，这种抽样方法往往比普通随机抽样方法更有效。

下图是按顺序对每 9 个元素进行一次采样，然后重复下去。

用Python实现的话可以直接在循环体中设置step即可。

population = 100  
step = 5  
sample = [element for element in range(1, population, step)]  
print (sample)

5.多级采样(Multistage sampling)

在多阶段采样下，我们将多个采样方法一个接一个地连接在一起。比如，在第一阶段，可以使用聚类抽样从总体中选择集群，然后第二阶段再进行随机抽样，从每个集群中选择元素以形成最终集合。

Python代码复用了上面聚类抽样，只是在最后一步再进行随机抽样即可。

import numpy as np  
clusters=5  
pop_size = 100  
sample_clusters=2  
sample_size=5  
# 间隔为 20, 从 1 到 5 依次分配集群100个样本的聚类 ID，这一步已经假设聚类完成  
cluster_ids = np.repeat([range(1,clusters+1)], pop_size/clusters)  
# 随机选出两个聚类的 ID  
cluster_to_select = random.sample(set(cluster_ids), sample_clusters)  
# 提取聚类 ID 对应的样本  
indexes = [i for i, x in enumerate(cluster_ids) if x in cluster_to_select]  
# 提取样本序号对应的样本值  
cluster_associated_elements = [el for idx, el in enumerate(range(1, 101)) if idx in indexes]  
# 再从聚类样本里随机抽取样本  
print (random.sample(cluster_associated_elements, sample_size))