%matplotlib inline

import nltk
from nltk.corpus import words
from functools import reduce

# Filter out words that don't need to be considered
filtered_list = list(filter(lambda x: "ie" in x or "ei" in x, words.words()))

# Apply the rule "i before e except after c" and see if it is true or false
def the_rule(word):
    if 'cie' in word:
        return False
    if 'cei' in word:
        word = word.replace("cei", '')
    if 'ei' in word:
        return False
    return True

ruling_dict = dict(map(lambda word: (word, the_rule(word)), filtered_list))

# Count the total number of rule followers and breakers
true_count = reduce(lambda count, val: count + (val == True), list(ruling_dict.values()), 0)
false_count = len(ruling_dict) - true_count

# Plot our discoveries
import matplotlib.pyplot as plt
import numpy as np

dictionary = plt.figure()

D = {u'True':true_count, u'False':false_count}

plt.bar(range(len(D)), D.values(), align='center')
plt.ylabel("Word Count")
plt.yticks(np.arange(0, len(ruling_dict), 750))
plt.xticks(range(len(D)), D.keys())
plt.title('Word follows the rule "i before e except after c"')