advanced code snippet search

We Recommend

Learning Python Learning Python
The authors of Learning Python show you enough essentials of the Python scripting language to enable you to begin solving problems right away, then reveal more powerful aspects of the language one at a time. This approach is sure to appeal to programmers and system administrators who have urgent problems and a preference for learning by semi-guided experimentation.

Posted By

stagger on 02/24/08


Tagged

textmate python ttm thesis


Versions (?)


Transport Matrix Generator

Published in: Python 

Plain TextExpand 
  1. import csv
  2. import time
  3.  
  4. reader = csv.reader(open("2007.csv", "rb"), dialect="excel", delimiter=";")
  5. opcodes = csv.reader(open("opcodes.csv", "rb"), dialect="excel", delimiter=";")
  6. writer = csv.writer(file("matrix.csv", "w"), dialect="excel", delimiter=";") 
  7.  
  8. # initialise transport matrix
  9. # column 50, row 2: matrix[50][2]
  10. matrix = []
  11. heading = ['']
  12. index = {}
  13. op = {}
  14. # i is the column index
  15. i = 0
  16. for row in opcodes:
  17.     code = row[0]
  18.     index[code] = i
  19.     op[i] = code
  20.     matrix.append([code])
  21.     heading.append(code)
  22.     i += 1
  23. for col in matrix:
  24.     for j in range(i):
  25.         col.append(0)
  26.  
  27. # add OP codes from CSV file
  28. for row in reader:
  29.     # parse current row into sequence[] with elements like: {'code':code, 'depth':depth}
  30.     sequence = []
  31.     code = ''
  32.     depth = 0
  33.     for char in row[0]:
  34.         if char in ' )' and not code == '':
  35.             def add(code, depth):
  36.                 step = {'code': code, 'depth': depth}
  37.                 sequence.append(step)
  38.             add(code, depth)
  39.             code = ''
  40.  
  41.         if char == '(':
  42.             depth += 1
  43.         elif char == ')':
  44.             depth -= 1
  45.         elif char != ' ':
  46.             code += char
  47.     add(code, depth)
  48.  
  49.     # add sequence to matrix
  50.     stack = []
  51.     quantity = int(row[1])
  52.     prev_depth = sequence[0]['depth']
  53.     prev_code = ''
  54.     for step in sequence:
  55.         depth = step['depth']
  56.         code = step['code']
  57.         #print '%s:%d nach %s:%d' % (code, depth, prev_code, prev_depth)
  58.         if (depth < prev_depth) and not (stack == []) and (depth == stack[-1]['depth']):
  59.             top = stack.pop()
  60.             matrix[index[top['code']]][index[code]+1] += quantity
  61.         elif depth > prev_depth:
  62.             stack.append({'code': prev_code, 'depth': prev_depth})
  63.         if prev_code and code and not (depth > prev_depth):
  64.             matrix[index[prev_code]][index[code]+1] += quantity
  65.         prev_depth = depth
  66.         prev_code = code
  67.  
  68. writer.writerow(heading)
  69. count = 1
  70. for column in matrix:
  71.     writer.writerow(column)
  72.  
  73. time.sleep(1)

Report this snippet 

You need to login to post a comment.