Saturday morning fun.

word_magic.py

import sys, os.path, csv

# Run me like: python word_magic.py inputfile.csv output_dir

KEEP_WORD = True

# We're being lazy here - no main for us, it's Saturday
in_filename = sys.argv[1]
out_dir     = os.path.abspath(sys.argv[2])

f = open(in_filename, 'rb')
reader = csv.reader(f)
lines = [x for x in reader]
f.close()

# Build a dictionary of the words, storing the lines we find for each
word_files = {}

for whole_line in lines:
    the_word = whole_line[0]
    if KEEP_WORD:
        # If KEEP_WORD is True, store the whole line.
        the_line = whole_line
    else:
        the_line = whole_line[1:]
    if not word_files.has_key(the_word):
        # Never seen this word before, setup a new output file for it
        word_files[the_word] = []
    # Add the line
    word_files[the_word].append(the_line)

# Now make our output files
for each_word in word_files.keys():
    f = open(os.path.join(out_dir, '%s.csv' % each_word), 'wb')
    writer = csv.writer(f, quoting=csv.QUOTE_MINIMAL)
    for each_entry in word_files[each_word]:
        writer.writerow(each_entry)
    f.close()

print "Done"