用着Python学量化Python之歌

和朋友赌骰子如何赢 - Python,数据分析及绘图,pygal

2017-04-19  本文已影响222人  牧羊的小狼

pygal 是非常不错的 python 可视化包,它可以生成矢量图,从而在不同尺寸的屏幕上都可以获得良好的显示效果。具体安装使用请参考官方文档。今天小狼举个投骰子的栗子,看一下不同点数的概率分布情况。源码请移步小狼 GitHub 下的 roll_dice_pygal 文件夹。

最常见的骰子是 6 个面的,点数分别从 1 至 6, 玩法上可以只投一个骰子,同时投两个骰子或者更多,在这里我们分别计算投一个,投两个和投三个的情况。

1. 投一个骰子的点数分布

每次投掷骰子的点数可以用 random 中的 randint 来实现,然后写两个函数方便待会儿复用:roll_dice() 用来模拟每次投掷骰子的点数,frequencies() 用来计算每个点数的频率。

from random import randint

def roll_dice():
    results = []
    for num in range(times):
        result = randint(1, sides)
        results.append(result)
    return results

def frequencies(x):
    results = []
    for num in range(1, sides+1):
        result = x.count(num)
        results.append(result)
    return results

设置骰子的面数为 6,投掷次数为 100 次:

sides = 6
times = 100

存储投掷结果(不放心的话可以 print 出来检查一下代码),计算每个点数的频率,把频率打印出来:

roll_results = roll_dice()
frequencies = frequencies(roll_results)
print(frequencies)

运行一下代码,得到频率的结果:

[17, 19, 15, 16, 13, 20]

上面的列表返回的分别是骰子 1 - 6 的频率结果,例如投掷 100 次,点数为 1 的次数共计是 17 次。看上去没什么问题,是时候该 pygal 上场了!

import pygal

freq_visual = pygal.Bar()

我们导入了 pygal,然后决定用柱状图来显示结果。接下来我们顺便对即将显示的图表做一些优化,让它更易读:

freq_visual.title = 'Rolling Results of 100 times'
freq_visual.x_labels = [str(x) for x in range(1, 7)]
freq_visual.x_title = 'Results'
freq_visual.y_title = 'Frequency'

接下来导入参数并且绘制图表:

freq_visual.add('6-side Dice', frequencies)
freq_visual.render_to_file('dice.svg')

我们得到了一个 .svg 格式的图,可以用浏览器打开查看,该格式也可以用 Adobe Illustrator 打开编辑。

dice_100.png

你和朋友玩了 100 次,你朋友赌 6 赢了 20 次,你赌 5 赢了 13 次,乍看下去似乎不是等概率事件?其实是因为投掷的次数太少了!你再和他玩个一万次,一百万次看看:

dice_10000.png dice_1000000.png

呵呵,呵呵呵。

2. 两个骰子的点数分布

现在我们同时投掷两个骰子,每次的结果等于两个骰子点数之和,稍稍修改一下计算频率的函数:

def frequencies(x):
    results = []
    for num in range(2, sides*2+1):
        result = x.count(num)
        results.append(result)
    return results

把第二个骰子的点数纳入到结果中:

roll_results_1 = roll_dice()
roll_results_2 = roll_dice()
roll_results = [roll_results_1[i]+roll_results_2[i] for i in range(times)]

记得改一下 x 轴坐标:

freq_visual.x_labels = [str(x) for x in range(2, 13)]

来看看模拟投掷 100 次,10,000 次和 1,000,000 次的结果:

dice_two_100.png dcie_two_10000.png dice_two_1000000.png

额,看来还是赌 7 是王道,但是前 100 次的输赢就看你的人品了。。。

3. 三个骰子的点数分布

现在我们同时投掷三个骰子,每次的结果等于三个骰子点数之和,再稍稍修改一下计算频率的函数:

def frequencies(x):
    results = []
    for num in range(3, sides*3+1):
        result = x.count(num)
        results.append(result)
    return results

把第三个骰子的点数纳入到结果中:

roll_results_1 = roll_dice()
roll_results_2 = roll_dice()
roll_results_3 = roll_dice()
roll_results = [roll_results_1[i]+roll_results_2[i]+roll_results_3[i] for i in range(times)]

最后记得改一下 x 轴坐标:

freq_visual.x_labels = [str(x) for x in range(3, 19)]

模拟投掷 100 次,10,000 次和 1,000,000 次的结果:

dice_three_100.png dice_three_10000.png dice_three_1000000.png

总之,相信你赌 10 和 11 在多数情况下都是能够完胜你那个赌 1 和 18 的朋友的。不过,小赌怡情,大赌伤身哦!

上一篇下一篇

猜你喜欢

热点阅读