Welcome to the Treehouse Community

Want to collaborate on code errors? Have bugs you need feedback on? Looking for an extra set of eyes on your latest project? Get support with fellow developers, designers, and programmers of all backgrounds and skill levels here with the Treehouse Community! While you're at it, check out some resources Treehouse students have shared here.

Looking to learn something new?

Treehouse offers a seven day free trial for new students. Get access to thousands of hours of content and join thousands of Treehouse students and alumni in the community today.

Start your free trial

Python Python Collections (Retired) Dictionaries Word Count

how to create a function named word count that takes a string and return a dictionary

how to create a function named word count that takes a string and return a dictionary

word_count.py
# E.g. word_count("I am that I am") gets back a dictionary like:
# {'i': 2, 'am': 2, 'that': 1}
# Lowercase the string to make it easier.
# Using .split() on the sentence will give you a list of words.
# In a for loop of that list, you'll have a word that you can
# check for inclusion in the dict (with "if word in dict"-style syntax).
# Or add it to the dict with something like word_dict[word] = 1.
def word_count(my_string):
    string_dict={}
    for wd in my_string.split():

        if wd in string_dict[wd]:
            string_dict[wd]+=1
    else:
        string_dict[wd]=1
    return string_dict

2 Answers

William Li
PLUS
William Li
Courses Plus Student 26,868 Points

Your code only has couple minor problems

  1. incorrect indentation on some lines
  2. if wd in string_dict[wd]: should be if wd in string_dict:.
def word_count(my_string):
    string_dict={}
    for wd in my_string.split():

        if wd in string_dict:
            string_dict[wd]+=1
        else:
            string_dict[wd]=1
    return string_dict
Paul Bentham
Paul Bentham
24,090 Points

Hi,

This question has already been answered here:

https://teamtreehouse.com/forum/splitstring-not-defined-whats-wrong-here-stuck

You need to set the values to your keys in string_dict to zero first.

William Li
William Li
Courses Plus Student 26,868 Points

No, you don't. When a word is found, it should be added to dictionary and set its value to 1. And running the for loop once is enough, no need to run it twice.

Paul Bentham
Paul Bentham
24,090 Points

You're correct. Thanks for that :)