python实现蓄水池抽样算法 (附完整源码)

源代码大师

于 2024-12-02 08:15:19 发布

阅读量292

点赞数 7

分类专栏： python算法完整教程文章标签： python 算法开发语言

不予转载，严禁转载，违者必纠。

本文链接：https://blog.csdn.net/it_xiangqiang/article/details/144178741

版权

python算法完整教程专栏收录该内容

815 篇文章 ¥29.90 ¥99.00

订阅专栏

超级会员免费看

python实现蓄水池抽样算法

蓄水池抽样算法（Reservoir Sampling）是一种在不知道数据总量（n）的情况下，从一个数据流中均匀随机地抽取k个样本的算法。这个算法特别适用于处理大型数据集或无限数据流，因为它只需要存储k个样本，无论数据流有多大。

下面是一个Python实现的完整代码，包含详细的注释和示例用法。

import random

def reservoir_sampling(stream, k):
    """
    从数据流stream中随机抽取k个样本，确保每个样本被选中的概率相同。

    参数:
        stream (iterable): 数据流，可以是任何可迭代对象。
        k (int): 要抽取的样本数量。

    返回:
        list: 抽取的k个样本。
    """
    reservoir = [