python的random模块及加权随机算法的python实现方法


Posted in Python onJanuary 04, 2017

random是用于生成随机数的,我们可以利用它随机生成数字或者选择字符串。

•random.seed(x)改变随机数生成器的种子seed。

一般不必特别去设定seed,Python会自动选择seed。

•random.random()    用于生成一个随机浮点数n,0 <= n < 1

•random.uniform(a,b)    用于生成一个指定范围内的随机浮点数,生成的随机整数a<=n<=b;

•random.randint(a,b)    用于生成一个指定范围内的整数,a为下限,b为上限,生成的随机整数a<=n<=b;若a=b,则n=a;若a>b,报错

•random.randrange([start], stop [,step])    从指定范围[start,stop)内,按指定基数递增的集合中获取一个随机数,基数缺省值为1

•random.choice(sequence)    从序列中获取一个随机元素,参数sequence表示一个有序类型,并不是一种特定类型,泛指list,tuple,字符串等

•random.shuffle(x[,random])    用于将一个列表中的元素打乱 (洗牌),会改变原始列表

•random.sample(sequence,k)    从指定序列中随机获取k个元素作为一个片段返回,不会改变原有序列

那么现在基础知识有了,我们来实现一个加权随机算法:

加权随机算法一般应用在以下场景:有一个集合S,里面比如有A,B,C,D这四项。这时我们想随机从中抽取一项,但是抽取的概率不同,比如我们希望抽到A的概率是50%,抽到B和C的概率是20%,D的概率是10%。一般来说,我们可以给各项附一个权重,抽取的概率正比于这个权重。那么上述集合就成了:

{A:5,B:2,C:2,D:1}

方法一:

最简单的方法可以这样:

把序列按权重值扩展成:lists=[A,A,A,A,A,B,B,C,C,D],然后random.choice(lists)随机选一个就行。虽然这样选取的时间复杂度是O(1),但是数据量一大,空间消耗就太大了。

# coding:utf-8
import random


def weight_choice(list, weight):
  """
  :param list: 待选取序列
  :param weight: list对应的权重序列
  :return:选取的值
  """
  new_list = []
  for i, val in enumerate(list):
    new_list.extend(val * weight[i])
  return random.choice(new_list)


if __name__ == "__main__":
  print(weight_choice(['A', 'B', 'C', 'D'], [5, 2, 2, 1]))

方法二:

比较常用的方法是这样:

计算权重总和sum,然后在1到sum之间随机选择一个数R,之后遍历整个集合,统计遍历的项的权重之和,如果大于等于R,就停止遍历,选择遇到的项。

还是以上面的集合为例,sum等于10,如果随机到1-5,则会在遍历第一个数字的时候就退出遍历。符合所选取的概率。

选取的时候要遍历集合,它的时间复杂度是O(n)。

# coding:utf-8
import random

list = ['A', 'B', 'C', 'D']


def weight_choice(weight):
  """
  :param weight: list对应的权重序列
  :return:选取的值在原列表里的索引
  """
  t = random.randint(0, sum(weight) - 1)
  for i, val in enumerate(weight):
    t -= val
    if t < 0:
      return i


if __name__ == "__main__":
  print(list[weight_choice([5, 2, 2, 1])])

方法三:

可以先对原始序列按照权重排序。这样遍历的时候,概率高的项可以很快遇到,减少遍历的项。(因为rnd递减的速度最快(先减去最大的数))

比较{A:5,B:2,C:2,D:1}和{B:2,C:2,A:5,D:1}

前者遍历步数的期望是5/10*1+2/10*2+2/10*3+1/10*4=19/10而后者是2/10*1+2/10*2+5/10*3+1/10*4=25/10。

这样提高了平均选取速度,但是原序列排序也需要时间。

先搞一个权重值的前缀和序列,然后在生成一个随机数t后,可以用二分法来从这个前缀和序列里找,那么选取的时间复杂度就是O(logn)了。

 

# coding:utf-8
import random
import bisect

list = ['A', 'B', 'C', 'D']


def weight_choice(weight):
  """
  :param weight: list对应的权重序列
  :return:选取的值在原列表里的索引
  """
  weight_sum = []
  sum = 0
  for a in weight:
    sum += a
    weight_sum.append(sum)
  t = random.randint(0, sum - 1)
  return bisect.bisect_right(weight_sum, t)


if __name__ == "__main__":
  print(list[weight_choice([5, 2, 2, 1])])

以上这篇python的random模块及加权随机算法的python实现方法就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
Python中的CURL PycURL使用例子
Jun 01 Python
python文件操作之目录遍历实例分析
May 20 Python
python 多线程实现检测服务器在线情况
Nov 25 Python
Django的信号机制详解
May 05 Python
Python数据分析之如何利用pandas查询数据示例代码
Sep 01 Python
浅谈python jieba分词模块的基本用法
Nov 09 Python
Python元字符的用法实例解析
Jan 17 Python
如何为Python终端提供持久性历史记录
Sep 03 Python
django实现类似触发器的功能
Nov 15 Python
pycharm 中mark directory as exclude的用法详解
Feb 14 Python
Python+Opencv身份证号码区域提取及识别实现
Aug 25 Python
python numpy中setdiff1d的用法说明
Apr 22 Python
python 实现红包随机生成算法的简单实例
Jan 04 #Python
Python 模板引擎的注入问题分析
Jan 01 #Python
python getopt详解及简单实例
Dec 30 #Python
浅谈编码,解码,乱码的问题
Dec 30 #Python
Python实现将数据库一键导出为Excel表格的实例
Dec 30 #Python
python脚本实现数据导出excel格式的简单方法(推荐)
Dec 30 #Python
利用python生成一个导出数据库的bat脚本文件的方法
Dec 30 #Python
You might like
正义联盟的终局之战《天启星战争》将成为DC动画宇宙的最后一部
2020/04/09 欧美动漫
PHP提取中文首字母
2008/04/09 PHP
解决file_get_contents无法请求https连接的方法
2013/12/17 PHP
php设计模式之职责链模式实例分析【星际争霸游戏案例】
2020/03/27 PHP
发布一个高效的JavaScript分析、压缩工具 JavaScript Analyser
2007/11/30 Javascript
javascript检查日期格式的函数[比较全]
2008/10/17 Javascript
javascript window.opener的用法分析
2010/04/07 Javascript
xml文档转换工具,附图表例子(hta)
2010/11/17 Javascript
select多选 multiple的使用示例
2014/06/16 Javascript
手机Web APP如何实现分享多平台功能
2016/08/19 Javascript
总结十个Angular.js由浅入深的面试问题
2016/08/26 Javascript
ionic隐藏tabs的方法
2016/08/29 Javascript
Vue.js教程之计算属性
2016/11/11 Javascript
简单实现jQuery级联菜单
2017/01/09 Javascript
jquery.cookie.js的介绍与使用方法
2017/02/09 Javascript
javascript计算对象长度的方法
2017/10/25 Javascript
浅谈es6中export和export default的作用及区别
2018/02/07 Javascript
JavaScript中的一些实用小技巧总结
2019/04/07 Javascript
Python利用Nagios增加微信报警通知的功能
2016/02/18 Python
Python tkinter模块弹出窗口及传值回到主窗口操作详解
2017/07/28 Python
TensorFlow如何实现反向传播
2018/02/06 Python
python实现朴素贝叶斯分类器
2018/03/28 Python
使用Python对微信好友进行数据分析
2018/06/27 Python
Python定义二叉树及4种遍历方法实例详解
2018/07/05 Python
Python推导式简单示例【列表推导式、字典推导式与集合推导式】
2018/12/04 Python
实例详解Python模块decimal
2019/06/26 Python
Django app配置多个数据库代码实例
2019/12/17 Python
详解django使用include无法跳转的解决方法
2020/03/19 Python
菲律宾票务网站:StubHub菲律宾
2018/04/21 全球购物
JD Sports丹麦:英国领先的运动时尚零售商
2020/11/24 全球购物
Python面试题:Python是如何进行内存管理的
2014/08/04 面试题
外语系大学生自荐信范文
2014/03/01 职场文书
学校花圃的标语
2014/06/18 职场文书
导游词之岳阳楼
2019/09/25 职场文书
Java实战之用Swing实现通讯录管理系统
2021/06/13 Java/Android
MySQL 数据类型详情
2021/11/11 MySQL