【数据分析面试】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

相关文章

Python将博客内容html导出为Markdown格式

《Python将博客内容html导出为Markdown格式》Python将博客内容html导出为Markdown格式,通过博客url地址抓取文章,分析并提取出文章标题和内容,将内容构建成html,再转... 目录一、为什么要搞?二、准备如何搞?三、说搞咱就搞!抓取文章提取内容构建html转存markdown

Python获取中国节假日数据记录入JSON文件

《Python获取中国节假日数据记录入JSON文件》项目系统内置的日历应用为了提升用户体验,特别设置了在调休日期显示“休”的UI图标功能,那么问题是这些调休数据从哪里来呢?我尝试一种更为智能的方法:P... 目录节假日数据获取存入jsON文件节假日数据读取封装完整代码项目系统内置的日历应用为了提升用户体验,

Python FastAPI+Celery+RabbitMQ实现分布式图片水印处理系统

《PythonFastAPI+Celery+RabbitMQ实现分布式图片水印处理系统》这篇文章主要为大家详细介绍了PythonFastAPI如何结合Celery以及RabbitMQ实现简单的分布式... 实现思路FastAPI 服务器Celery 任务队列RabbitMQ 作为消息代理定时任务处理完整

Python Websockets库的使用指南

《PythonWebsockets库的使用指南》pythonwebsockets库是一个用于创建WebSocket服务器和客户端的Python库,它提供了一种简单的方式来实现实时通信,支持异步和同步... 目录一、WebSocket 简介二、python 的 websockets 库安装三、完整代码示例1.

揭秘Python Socket网络编程的7种硬核用法

《揭秘PythonSocket网络编程的7种硬核用法》Socket不仅能做聊天室,还能干一大堆硬核操作,这篇文章就带大家看看Python网络编程的7种超实用玩法,感兴趣的小伙伴可以跟随小编一起... 目录1.端口扫描器:探测开放端口2.简易 HTTP 服务器:10 秒搭个网页3.局域网游戏:多人联机对战4.

使用Python实现快速搭建本地HTTP服务器

《使用Python实现快速搭建本地HTTP服务器》:本文主要介绍如何使用Python快速搭建本地HTTP服务器,轻松实现一键HTTP文件共享,同时结合二维码技术,让访问更简单,感兴趣的小伙伴可以了... 目录1. 概述2. 快速搭建 HTTP 文件共享服务2.1 核心思路2.2 代码实现2.3 代码解读3.

Python使用自带的base64库进行base64编码和解码

《Python使用自带的base64库进行base64编码和解码》在Python中,处理数据的编码和解码是数据传输和存储中非常普遍的需求,其中,Base64是一种常用的编码方案,本文我将详细介绍如何使... 目录引言使用python的base64库进行编码和解码编码函数解码函数Base64编码的应用场景注意

Python基于wxPython和FFmpeg开发一个视频标签工具

《Python基于wxPython和FFmpeg开发一个视频标签工具》在当今数字媒体时代,视频内容的管理和标记变得越来越重要,无论是研究人员需要对实验视频进行时间点标记,还是个人用户希望对家庭视频进行... 目录引言1. 应用概述2. 技术栈分析2.1 核心库和模块2.2 wxpython作为GUI选择的优

Python如何使用__slots__实现节省内存和性能优化

《Python如何使用__slots__实现节省内存和性能优化》你有想过,一个小小的__slots__能让你的Python类内存消耗直接减半吗,没错,今天咱们要聊的就是这个让人眼前一亮的技巧,感兴趣的... 目录背景:内存吃得满满的类__slots__:你的内存管理小助手举个大概的例子:看看效果如何?1.

Python+PyQt5实现多屏幕协同播放功能

《Python+PyQt5实现多屏幕协同播放功能》在现代会议展示、数字广告、展览展示等场景中,多屏幕协同播放已成为刚需,下面我们就来看看如何利用Python和PyQt5开发一套功能强大的跨屏播控系统吧... 目录一、项目概述:突破传统播放限制二、核心技术解析2.1 多屏管理机制2.2 播放引擎设计2.3 专