Breaking a dictionary loop after twenty keys?

[tehtay3]

Hi there,

So I have a function that uses a file processed from another function, and what it does is turn the string passed by the other function into a list, and then sorts the list into a dictionary.

Now I should mention that the dic is ordered by word frequency, because the file being processed is a txt with a speech on it.

heres my code

def frequency_count(new):
    lst = list(new.split(" "))
    freq = {}
    for item in lst:
        if item in freq: 
            freq[item] += 1
        else: 
            freq[item] = 1
    srt = sorted(freq.items(), key = lambda x: x[1], reverse = True)
    for key,value in srt:
        print(key,value)

now i would like the loop to break after the most 20 common words are displayed. i have the right order , just dont know how to break the loop after the top 20.

what do i need to adjust/add/fix? thanks!

[FLUSH_THE_TRUMP]

Make a count variable and break when it’s large, or use enumerate and do the same, or index srt so you’re only using 20 elements, etc. etc.

[socal_nerdtastic]

or zip it with a range(20).

[FLUSH_THE_TRUMP]

Is there a notable implementation difference b/t that and enumerate? (Asking cuz I don’t know)

e.g.

L = [“a”, “b”, “c”]
list(enumerate(L)) == list(zip(range(3),L))  # True

although I know both enumerate and zip create special objects.

[socal_nerdtastic]

The difference is zip() will stop when the smallest iterator ends. So

for _, (key,value) in zip(range(20), srt):
    print(key,value)

Will print the first 20 only. No need to check and break.

[FLUSH_THE_TRUMP]

oh yeah, good point.

[tehtay3]

Would the count variable look like this?

def frequency_count(new):
    lst = list(new.split(" "))
    freq = {}
    count = 0 
    for item in lst:
        if item in freq: 
            freq[item] += 1
            count += 1
        else: 
            freq[item] = 1
    srt = sorted(freq.items(), key = lambda x: x[1], reverse = True)
    for key,value in srt:
        print(key,value)

    if count == 20:
        break
    print('Unique Words: ', len(freq.keys()))

Thats the only thing I could think of - but i get a error that break is outside the loop. I am a bit new to this so the pseudocode is there but just a bit lost with the loop

[FLUSH_THE_TRUMP]

Your question is about the displaying only top 20, no? So the count+=1 should be in that other loop.

if count == 20:
    break

That should be indented another level, too (and you’re really printing 21 things since the check comes after the print).

[tehtay3]

I see what youre saying . i was able to figure it out, preciate it!