change to log coordinate & dot plot

最新推荐文章于 2022-09-13 01:31:35 发布

转载最新推荐文章于 2022-09-13 01:31:35 发布 · 567 阅读

python 专栏收录该内容

97 篇文章

订阅专栏

本文介绍如何利用Python编程实现硬币翻转的概率模拟，并通过图表展示不同翻转次数下正反面比例的变化。

#quote from MIT 'introduction to computation and programming using python, Revised'
import random
import pylab

def flipPlot(minExp, maxExp):
    """Assumes minExp and maxExp positive integers; minExp < maxExp
       Plots results of 2**minExp to 2**maxExp coin flips"""
    ratios = []
    diffs = []
    xAxis = []
    for exp in range(minExp, maxExp + 1):
        xAxis.append(2**exp)
    for numFlips in xAxis:
        numHeads = 0
        for n in range(numFlips):
            if random.random() < 0.5:
                numHeads += 1
        numTrials = numFlips - numHeads
        ratios.append(numHeads/float(numTrials))
        diffs.append(abs(numHeads - numTrials))
    pylab.title('Difference Between Heads and Trials')
    pylab.xlabel('Number of Flips')
    pylab.ylabel('Abs(#Heads - #Tails)')
    pylab.plot(xAxis, diffs, 'bo')
    pylab.semilogx()
    pylab.semilogy()
    pylab.figure()
    pylab.title('Heads/Tails Ratios')
    pylab.xlabel('Number of Flips')
    pylab.ylabel('#Heads/#Tails')
    pylab.plot(xAxis, ratios, 'bo')
  
random.seed(0)  
flipPlot(4, 20)
pylab.show()