Split data into multiple files

multiplexer.py

import sys

path = sys.argv[1]
ext = 'txt'
dir = 'output'
limit = 10000

limits = {}

with open(path) as file:
    for line in file:
        try:
            (country, keyword, count) = line.strip().split("\t")
        except:
            continue
        if len(country) != 2:
            continue
        if limits.get(country, 0) < limit:
            with open("%s/%s.%s" % (dir, country, ext), 'a') as output:
                output.write(keyword)
                output.write("\n")
            limits[country] = limits.get(country, 0) + 1