Word Lengths

Whiteboard: Easier
Concepts: Dictionaries, Loops
Download: word-lengths.zip
Solution: Word Lengths: Solution

Given a phrase, return dictionary keyed by word-length, with the value for each length being the set of words of that length.

For example:

>>> answer = word_lengths("cute cats chase fuzzy rats")

This should return {4: {'cute', 'cats', 'rats'}, 5: {'chase', 'fuzzy'}}, but since both dictionaries and sets are unordered, we can’t just check if it matches that exact string, so we’ll test more carefully:

>>> sorted(answer.keys())
[4, 5]

>>> answer[4] == {'cute', 'cats', 'rats'}
True

>>> answer[5] == {'chase', 'fuzzy'}
True

Punctuation should be considered part of a word, so you only need to split the string on whitespace:

>>> answer = word_lengths("Hi, I'm Balloonicorn")

>>> sorted(answer.keys())
[3, 12]

>>> answer[3] == {'Hi,', "I'm"}
True

>>> answer[12] == {"Balloonicorn"}

We’ve given you wordlengths.py, which contains the stub of a word_lengths function:

wordlengths.py

def word_lengths(sentence):
    """Get dictionary of word-length: {words}."""

Implement this function.