【数据分析面试】12. 随机抽取颜色球(Python random模块应用:choices()/choice()/sample())

本文主要是介绍【数据分析面试】12. 随机抽取颜色球(Python random模块应用:choices()/choice()/sample()),希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

在这里插入图片描述

题目 随机抽取颜色球

编写一个函数来模拟从罐子中抽取球的过程。球的颜色存储在名为jar的列表中,每个颜色对应球的数量存储在名为n_balls的列表中,且数量与颜色列表的索引对应。

示例:

输入:

jar = ['green', 'red', 'blue']  
n_balls = [1, 10, 2]  
# 1个绿色球  
# 10个红色球  
# 和2个蓝色球

输出

sample_multinomial(jar, n_balls) -> "blue"

函数执行后应随机返回罐中某个颜色的球,抽取的概率与每种颜色球的数量成正比。在示例中,因为红色球数量最多,所以抽到红色球的概率最高,但也有可能抽到绿色或蓝色球。

答案

答案代码

import random def sample_multinomial(jar, n_balls): # 根据n_balls计算每个颜色的权重 weights = [count / sum(n_balls) for count in n_balls] # 使用random.choices按权重随机选择一个颜色 chosen_color = random.choices(jar, weights=weights, k=1)[0] return chosen_color 

上面这个函数使用random.choices函数来模拟从罐子中抽取球的过程,其中每个颜色被选中的概率与其在n_balls列表中的计数成正比。

在这段代码中,我们首先计算了每个颜色被选中的权重,通过将每个颜色的球数除以总球数来得到。然后,我们使用random.choices函数,它根据提供的权重列表随机选择元素。k=1参数表示我们只想抽取一个球。函数返回抽取到的球的颜色。

每次运行这段代码时,你都会得到一个随机抽取的颜色,抽取的概率与n_balls列表中的计数成正比。

random.choices()函数介绍

random.choices()是Python标准库中的一个函数,用于从一个给定的集合中随机选择元素。

其中,

  • population是需要从中选择元素的集合,可以是列表、元组、字符串等。
  • weights是一个可选参数,用于指定每个元素的选择权重,如果不提供,则默认所有元素具有相同的权重。
  • k是一个整数,表示需要选择的元素数量,默认为1。

以下是一些使用random.choices()的例子:

import random# 创建列表jar
jar=[1,2,3,4,5]# 随机选择5个元素,每个元素被选中的概率相等。
print(random.choices(jar, k=5))# 随机选择5个元素,但只有数字3有被选中的权重,因此结果中只会包含数字3。
print(random.choices(jar, weights=[0,0,1,0,0], k=5))#使用累积权重来确定每个元素被选中的概率。
print(random.choices(jar, cum_weights=[1,1,1,1,1], k=5))

总的来说,random.choices()是一个非常灵活的函数,可以根据需要调整选择元素的权重,非常适合于需要根据概率分布进行随机选择的情况。

除了random.choices(),还有另外两个类似的函数——random.choice()random.sample()

random.choice()

random.choice() 函数用于从非空序列(如列表、元组或字符串)中随机选择一个元素。它假设所有元素被选中的概率是相同的,并且每次调用只返回一个元素。如果序列为空,random.choice() 会抛出一个 IndexError 异常。

示例:

import random    
colors = ['red', 'green', 'blue']  
chosen_color = random.choice(colors)  
print(chosen_color)  # 输出可能是 'red'、'green' 或 'blue' 中的一个
random.sample()

random.sample() 函数用于从序列中随机选择多个不重复的元素。它需要两个参数:一个序列和一个整数 k,表示要选择的元素数量。返回的是一个新列表,其中包含从原始序列中随机选择的不重复的 k 个元素。如果序列中的元素数量少于 k,random.sample() 会抛出一个 ValueError 异常。

示例:

import random    
numbers = [1, 2, 3, 4, 5, 6]  
sampled_numbers = random.sample(numbers, 3)  
print(sampled_numbers)  
# 输出可能是 [1, 3, 5]、[2, 4, 6] 等包含三个不重复数字的列表

总结一下它们之间的主要区别:

  1. 返回值的数量和类型random.choice() 返回一个随机选择的元素,而 random.sample() 返回一个包含多个随机选择元素的列表。
  2. 是否允许重复random.choice() 允许在多次调用中重复选择相同的元素,而 random.sample() 确保所选元素在返回的列表中不重复。
  3. 错误处理:当序列为空时,random.choice() 会抛出 IndexError,而 random.sample() 在请求的样本大小超过序列长度时会抛出 ValueError
  4. 使用场景:当你只需要一个随机元素时,使用 random.choice();当你需要多个不重复的随机元素时,使用 random.sample()

其他常用Random函数汇总

常用的函数有以下几种:

  1. 生成随机浮点数
    random.random():返回一个范围在[0,1)内的随机浮点数。
    random.uniform(a, b):返回一个范围在[a, b)内的随机浮点数。
  2. 生成随机整数
    random.randint(a, b):返回一个范围在[a, b)内的随机整数,包含a和b的整数部分。
    random.randrange(a, b, step):从指定的范围内,按照给定的基数和步长值获取一个随机数值。
  3. 生成正态分布的随机数
    random.normalvariate(mean, stddev):返回满足指定均值(mean)和标准差(stddev)的正态分布(高斯分布)的概率密度随机数。
  4. 从序列中选择元素
    random.choice(seq):从非空序列中随机选择一个元素。
    random.sample(seq, k):从给定的总体序列或集合中随机选取k个唯一的元素。
  5. 打乱序列
    random.shuffle(x):将序列x中的元素打乱,随机排序。
  6. 设置随机种子
    random.seed([x]):初始化伪随机数生成器的种子值;如果未传入参数,则使用系统当前时间作为种子。
    在这里插入图片描述

这篇关于【数据分析面试】12. 随机抽取颜色球(Python random模块应用:choices()/choice()/sample())的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/882532

相关文章

C语言中位操作的实际应用举例

《C语言中位操作的实际应用举例》:本文主要介绍C语言中位操作的实际应用,总结了位操作的使用场景,并指出了需要注意的问题,如可读性、平台依赖性和溢出风险,文中通过代码介绍的非常详细,需要的朋友可以参... 目录1. 嵌入式系统与硬件寄存器操作2. 网络协议解析3. 图像处理与颜色编码4. 高效处理布尔标志集合

如何使用 Python 读取 Excel 数据

《如何使用Python读取Excel数据》:本文主要介绍使用Python读取Excel数据的详细教程,通过pandas和openpyxl,你可以轻松读取Excel文件,并进行各种数据处理操... 目录使用 python 读取 Excel 数据的详细教程1. 安装必要的依赖2. 读取 Excel 文件3. 读

Python的time模块一些常用功能(各种与时间相关的函数)

《Python的time模块一些常用功能(各种与时间相关的函数)》Python的time模块提供了各种与时间相关的函数,包括获取当前时间、处理时间间隔、执行时间测量等,:本文主要介绍Python的... 目录1. 获取当前时间2. 时间格式化3. 延时执行4. 时间戳运算5. 计算代码执行时间6. 转换为指

利用Python调试串口的示例代码

《利用Python调试串口的示例代码》在嵌入式开发、物联网设备调试过程中,串口通信是最基础的调试手段本文将带你用Python+ttkbootstrap打造一款高颜值、多功能的串口调试助手,需要的可以了... 目录概述:为什么需要专业的串口调试工具项目架构设计1.1 技术栈选型1.2 关键类说明1.3 线程模

Python ZIP文件操作技巧详解

《PythonZIP文件操作技巧详解》在数据处理和系统开发中,ZIP文件操作是开发者必须掌握的核心技能,Python标准库提供的zipfile模块以简洁的API和跨平台特性,成为处理ZIP文件的首选... 目录一、ZIP文件操作基础三板斧1.1 创建压缩包1.2 解压操作1.3 文件遍历与信息获取二、进阶技

Python Transformers库(NLP处理库)案例代码讲解

《PythonTransformers库(NLP处理库)案例代码讲解》本文介绍transformers库的全面讲解,包含基础知识、高级用法、案例代码及学习路径,内容经过组织,适合不同阶段的学习者,对... 目录一、基础知识1. Transformers 库简介2. 安装与环境配置3. 快速上手示例二、核心模

Python正则表达式语法及re模块中的常用函数详解

《Python正则表达式语法及re模块中的常用函数详解》这篇文章主要给大家介绍了关于Python正则表达式语法及re模块中常用函数的相关资料,正则表达式是一种强大的字符串处理工具,可以用于匹配、切分、... 目录概念、作用和步骤语法re模块中的常用函数总结 概念、作用和步骤概念: 本身也是一个字符串,其中

Python使用getopt处理命令行参数示例解析(最佳实践)

《Python使用getopt处理命令行参数示例解析(最佳实践)》getopt模块是Python标准库中一个简单但强大的命令行参数处理工具,它特别适合那些需要快速实现基本命令行参数解析的场景,或者需要... 目录为什么需要处理命令行参数?getopt模块基础实际应用示例与其他参数处理方式的比较常见问http

python实现svg图片转换为png和gif

《python实现svg图片转换为png和gif》这篇文章主要为大家详细介绍了python如何实现将svg图片格式转换为png和gif,文中的示例代码讲解详细,感兴趣的小伙伴可以跟随小编一起学习一下... 目录python实现svg图片转换为png和gifpython实现图片格式之间的相互转换延展:基于Py

Python中的getopt模块用法小结

《Python中的getopt模块用法小结》getopt.getopt()函数是Python中用于解析命令行参数的标准库函数,该函数可以从命令行中提取选项和参数,并对它们进行处理,本文详细介绍了Pyt... 目录getopt模块介绍getopt.getopt函数的介绍getopt模块的常用用法getopt模