Anagram data structure

anagram.py

from collections import defaultdict

words = ['eat', 'tea', 'ate', 'time'] # ... get words from somewhere else
data = defaultdict(set)

for word in words:
  data[''.join(sorted(word))].add(word)

def get_anagrams_for(word):
  words = data[''.join(sorted(word))]
  words.remove(word)
  return list(words)

if __name__ == '__main__':
  print get_anagrams_for('tea') # prints ['ate', 'eat']

mines not as good - but still fun.

from itertools import groupby
words = ['eat', 'tea', 'ate', 'time']
words.sort(key=sorted)
for sorted_, words in groupby(words, key=sorted):
... print list(words)
...
['eat', 'tea', 'ate']
['time']