Group words by their first letter

group_by_letter.py

#!/usr/bin/env python

"""Group words by their first letter"""

from collections import defaultdict

def group_by_letter(words):
    buckets = defaultdict(lambda:[])
    for word in words:
        buckets[word[0].lower()].append(word)
    return buckets

if __name__ == '__main__':
    print group_by_letter(['narragansett','brooklyn lager','magic hat','dog fish head','shock top','ten penny','bass'])
    # Output: defaultdict(<function <lambda> at 0x7fc83416b2a8>, {'b': ['bass', 'brooklyn lager'], 'd': ['dog fish head'], 'm': ['magic hat'], 'n':['narragansett'], 's': ['shock top'], 't': ['ten penny']})