In [30]:
import re
import sys

Export top 30 trip modes to geosheets


In [31]:
colors = ((5, 'red'), (15, 'blue'), (30, 'black'))

with open('results/trips_modes.txt', 'r') as ttf, open('modes_culler_geosheets.csv', 'w+') as outf:
    outf.write('Location\tType\tMode\tColor\tConfidence\n')
    count, color_count = 0, 1
    current_color = colors[0]
    for line in ttf:
        found = re.search(r'.*\'(.*) (.*)\'.*\'(.*) (.*)\'.*\'(.*)\'.*\[(.*)\].*', line, re.M|re.I)
        if found is not None:
            if count == 30:
                break
            if count == current_color[0]:
                current_color = colors[color_count]
                color_count += 1
            outf.write('%s,%s | %s,%s\tline\t%s\t%s\t%s\n' % (found.group(2), found.group(1), found.group(4), found.group(3), found.group(5), current_color[1], found.group(6)))
            count += 1

Export top 30 trip times to geosheets


In [32]:
with open('results/trips_times.txt', 'r') as ttf, open('times_culler_geosheets.csv', 'w+') as outf:
    outf.write('Location\tType\tTime\tColor\tConfidence\n')
    count, color_count = 0, 1
    current_color = colors[0]
    for line in ttf:
        found = re.search(r'.*\'(.*) (.*)\'.*\'(.*) (.*)\'.*\'(.*)\'.*\'.*\'.*\[(.*)\].*', line, re.M|re.I)
        if found is not None:
            if count == 30:
                break
            if count == current_color[0]:
                current_color = colors[color_count]
                color_count += 1
            outf.write('%s,%s | %s,%s\tline\t%s\t%s\t%s\n' % (found.group(2), found.group(1), found.group(4), found.group(3), found.group(5), current_color[1], found.group(6)))
            count += 1

In [ ]: