Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- #!/usr/bin/python
- import csv
- import glob
- import os
- import sys
- input_path = sys.argv[1]
- output_file = sys.argv[2]
- my_columns = ['Street', 'Booked?', 'Notes']
- for input_file in glob.glob(os.path.join(input_path, '*.csv')):
- with open(input_file, 'rU') as csv_in_file:
- filereader = csv.reader(csv_in_file, delimiter=',')
- filewriter = csv.writer(output_file, delimiter=',')
- header = next(filereader, None)
- header_list_output =[]
- for index_value in range(len(header)):
- if header[index_value] in my_columns:
- header_list_output.append(header[index_valuex`])
- filewriter.writerow(header_list_output)
- for row in filereader:
- row_list_output=[]
- for index_value in range(len(header)):
- if header[index_value] in my_columns:
- row_list_output.append(row[index_value])
- filewriter.writerow(row_list_output)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement