import sys import random distribs = [ 'a:0.6 the:0.2 :0.2', 'the:0.7 a:0.2 :0.1', 'the:0.9 :0.1', 'the:0.3 be:0.2 to:0.15 of:0.15 and:0.05 a:0.05 in:0.05 :0.05', ] for line in sys.stdin: ctx = line.split('\t')[6:] i = random.randint(0, len(distribs) - 1) print(distribs[i])