Welcome to the Treehouse Community

The Treehouse Community is a meeting place for developers, designers, and programmers of all backgrounds and skill levels to get support. Collaborate here on code errors or bugs that you need feedback on, or asking for an extra set of eyes on your latest project. Join thousands of Treehouse students and alumni in the community today. (Note: Only Treehouse students can comment or ask questions, but non-students are welcome to browse our conversations.)

Looking to learn something new?

Treehouse offers a seven day free trial for new students. Get access to thousands of hours of content and a supportive community. Start your free trial today.

Python Python Collections (Retired) Dictionaries Word Count

My code works when I run it in PyCharm but I get an error 'Didn't get the right count for some words'.

'''def word_count(string): string2 = string.lower() dict = {} list = string2.split() for element in list: dict.update({element : string2.count(element)}) return dict

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(string):
    string2 = string.lower()
    dict = {}
    list = string2.split()
    for element in list:
      dict.update({element : string2.count(element)})
    return dict

3 Answers

Chris Freeman
MOD
Chris Freeman
Treehouse Moderator 67,989 Points

Your code is close. You are counting on string2 which is not divided into words and may give false matches. Change string2 to list:

def word_count(string):
    string2 = string.lower()
    dict = {}
    list = string2.split()
    for element in list:
      dict.update({element : list.count(element)})
    return dict
Chris Freeman
Chris Freeman
Treehouse Moderator 67,989 Points

Additional feedback: It is best not to use the names of built-in functions (list, dict) as variable names. It worked in this code, but can have bad consequences in more complicated code.

Hi Chris, thanks for the prompt feedback. Maybe I am missing something but I thought that list = string2.split() would split string2 into words? This list of words would be in 'list' then the count method is on the list (list.count(element). Please can you let me know where I am going wrong. Thank you.

Chris Freeman
Chris Freeman
Treehouse Moderator 67,989 Points

I thought that list = string2.split() would split string2 into words?

Correct.

This list of words would be in 'list' then the count method is on the list (list.count(element)).

Also correct.

Please can you let me know where I am going wrong. Thank you.

The difference is in your original code you had:

string2.count(element)

Instead of

list.count(element)

Ah, yes. I see that now! These forums are brilliant for people learning. It's often the little things that lead to hang-ups. Thanks a lot.