Harvard:Biophysics 101/2007/Notebook:Michael Wang/2007-2-8

Code:

#!/usr/bin/env python

import random
import string

#countRepeats counts the number of sequences in a single trial passed as a string
#and then counts the consecutive string occurances and adds them to the appropriate
#count matrix
def countRepeats(trial, search_char, matrix):    
    for i in range (10):
        substr = ''.join([search_char for n in range (i+1)]) #create the substring to search for of appropriate length
        count = 0
        pos = trial.find(substr,0)
        while not pos == -1:
            count = count + 1
            pos = trial.find(substr, pos +1)
        matrix[i][count]+=1  #add one to the appropriate element in the 2d matrix

        
#displayCounts displays the data stored in the matrix and calculates total occurances
def displayCounts (character, matrix):
    print
    print
    Title = "Number of trials with n occurances of the given consecutive string of "+character
    print Title.rjust(80)
    print "n".rjust(45)
    print "String".rjust(len(matrix)),
    for m in range(len(matrix[0])):
        print repr(m).rjust(6),
    print "Total*".rjust(8)
    totals = [0,0,0,0,0,0,0,0,0,0]
    for i in range(len(matrix)):
        substr = ''.join([character for n in range (i+1)])
        print substr.rjust(len(matrix)),
        for j in range(len(matrix[i])):
            print repr(matrix [i][j]).rjust (6),
        total = 0
        for k in range(len(matrix [i])):
            total+=matrix[i][k]*k
        print repr(total).rjust(7)
        totals[i]=total
    print "*Total refers to total occurances of a particular string"
    return totals
    

zero_list = range(10)
for i in range(10):
    zero_list[i]=0;
print zero_list;
#there must be a better way to do this....
#ARGH!  I originally intended to use zero_string below, but apparently it passes a pointer instead of a value...

count_of_H_counts = []
count_of_T_counts = []
for j in range (10):
    count_of_H_counts.append ([0,0,0,0,0,0,0,0,0,0,0]) #wanted to use zero_string here
    count_of_T_counts.append ([0,0,0,0,0,0,0,0,0,0,0]) #and here

for k in range (10000):  #This for loop controls the number of trials performed
                         #where each trial is ten tosses
    random.seed()

    #populating the trial    
    temp = ""
    for l in range (10):
        if random.random()<0.5:
            temp = temp + "H"
        else:
            temp = temp + "T"

    #call the countRepeats function to process the new temp trial and add to H/T matrix    
    countRepeats(temp, 'H',count_of_H_counts)
    countRepeats(temp, 'T',count_of_T_counts)

#call displaycounts to display the tables and calculate totals
Htotals=displayCounts ('H',count_of_H_counts)
Ttotals=displayCounts ('T',count_of_T_counts)

#display the totals
print
print
print "Total Counts"
for i in range(len(Htotals)):
    print repr(i+1).rjust(5),"|",Htotals[i]+Ttotals[i]

Output:

          Number of trials with n occurances of the given consecutive string of H
                                            n
    String      0      1      2      3      4      5      6      7      8      9     10   Total*
         H      5    102    447   1188   2065   2475   2040   1123    465     79     11   49837
        HH   1429   2356   2359   1775   1097    613    251     95     14     11      0   22234
       HHH   5025   2314   1366    739    339    148     44     14     11      0      0    9809
      HHHH   7578   1309    655    274    115     44     14     11      0      0      0    4282
     HHHHH   8922    638    256    115     44     14     11      0      0      0      0    1807
    HHHHHH   9560    256    115     44     14     11      0      0      0      0      0     729
   HHHHHHH   9816    115     44     14     11      0      0      0      0      0      0     289
  HHHHHHHH   9931     44     14     11      0      0      0      0      0      0      0     105
 HHHHHHHHH   9975     14     11      0      0      0      0      0      0      0      0      36
HHHHHHHHHH   9989     11      0      0      0      0      0      0      0      0      0      11
*Total refers to total occurances of a particular string


         Number of trials with n occurances of the given consecutive string of T
                                            n
    String      0      1      2      3      4      5      6      7      8      9     10   Total*
         T     11     79    465   1123   2040   2475   2065   1188    447    102      5   50163
        TT   1371   2312   2380   1806   1156    595    239    110     26      5      0   22546
       TTT   4929   2355   1444    707    345    141     48     26      5      0      0    9959
      TTTT   7556   1348    639    277    101     48     26      5      0      0      0    4292
     TTTTT   8941    627    252    101     48     26      5      0      0      0      0    1786
    TTTTTT   9568    252    101     48     26      5      0      0      0      0      0     727
   TTTTTTT   9820    101     48     26      5      0      0      0      0      0      0     295
  TTTTTTTT   9921     48     26      5      0      0      0      0      0      0      0     115
 TTTTTTTTT   9969     26      5      0      0      0      0      0      0      0      0      36
TTTTTTTTTT   9995      5      0      0      0      0      0      0      0      0      0       5
*Total refers to total occurances of a particular string


Total Counts
    1 | 100000
    2 | 44780
    3 | 19768
    4 | 8574
    5 | 3593
    6 | 1456
    7 | 584
    8 | 220
    9 | 72
   10 | 16