Advertisement
Guest User

Change column names

a guest
Aug 29th, 2015
32
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 0.65 KB | None | 0 0
  1. import sys
  2. import os
  3.  
  4. folder = sys.argv[1]
  5. for path in os.listdir(folder):
  6.  
  7.     wiki_db = path.split('.')[0]
  8.     full_path = os.path.join(folder, path)
  9.  
  10.     headers = None
  11.     values = []
  12.  
  13.     with open(full_path, 'r') as input_file:
  14.         for line in input_file:
  15.             if headers is None:
  16.                 headers = line
  17.             else:
  18.                 values.append(line)
  19.  
  20.     date_header, value_header = headers.strip().split('\t')
  21.     headers = date_header + '\t' + wiki_db + '\n'
  22.  
  23.     with open(full_path, 'w') as output_file:
  24.         output_file.write(headers)
  25.         for line in values:
  26.             output_file.write(line)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement