def plotN(N,p1,p2): tosses1 = generateTossesStatistics(N,p1) tosses2 = generateTossesStatistics(N,p2) tosses1X = map(lambda x: x[0], tosses1) tosses1Y = map(lambda x: x[1], tosses1) tosses2X = map(lambda x: x[0], tosses2) tosses2Y = map(lambda x: x[1], tosses2) ''' (X,Y) (#Tosses,#heads,#tails) ''' #plt.plot([1,2,3,4],[1,4,9,16],'ro') #plt.plot([1,2,3,4],[1,2,3,4],'ro',[1,2,3,4],[1,4,9,16],'bs') plt.plot(tosses1X,tosses1Y,'ro--',[0,N+1],[p1,p1],'ro-',tosses2X,tosses2Y,'bs--',[0,N+1],[p2,p2],'bs-') plt.axis([0, N+1, 0, 1]) plt.ylabel('average heads probability for each coin') #plt.show() s = "plot N_%d p1_%0.3f p2_%0.3f.png" %(N,p1,p2) #plt.savefig('plot'+ str(N)+"_p1_" +str(p1)+"_p2_" +str(p2)+'.png') '''plt.annotate('local max', xy=(10, 10), xytext=(3, 1.5), arrowprops=dict(facecolor='black', shrink=0.05), ) ''' plt.savefig(s)
def testgenerateTossesStatistics(self): N = 10 tosses = generateTossesStatistics(N,0.8) for x in tosses: print x '''