assignment-2b.py

#!/usr/bin/env python
import random
from letters import file2pairs
import sys

file = sys.argv[1]
prev_letter = sys.argv[2]
letters_dict = file2pairs(file)
prob = letters_dict[prev_letter]
p_value = prob.values()
support = 'abcdefghijklmnopqrstuvwxyz '
answer = random.choices(support,p_value)
print(answer[0])