anagrams.py

# Another from Programming Pearls (again, my attempt before reading
# the rest of the chapter). Given a dictionary and a word, find all
# anagrams of the word in the dictionary.

words = (x.strip().lower() for x in open("/usr/share/dict/words"))
lookup = {}
for word in words:
    letters = "".join(sorted(list(word)))
    lookup.setdefault(letters, []).append(word)

def anagrams(word):
    letters = "".join(sorted(list(word)))
    return lookup.get(letters, [])

print anagrams("pots")