lib/exercises-in-programming-style
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Make the last function also a generator, so that the explanation flows better

Browse Source
This commit is contained in:
Crista Lopes
2014-01-01 12:56:15 -08:00
parent 1fe7f0c9c7
commit ff435c3351
1 changed files with 9 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
26-lazy-rivers/tf-26.py
View File

@@ -31,15 +31,18 @@ def non_stop_words(filename):
yield w yield w
def count_and_sort(filename): def count_and_sort(filename):
freqs = {} freqs, i = {}, 1
for w in non_stop_words(filename): for w in non_stop_words(filename):
freqs[w] = 1 if w not in freqs else freqs[w]+1 freqs[w] = 1 if w not in freqs else freqs[w]+1
return sorted(freqs.iteritems(), key=operator.itemgetter(1), reverse=True) if i % 5000 == 0:
yield sorted(freqs.iteritems(), key=operator.itemgetter(1), reverse=True)
i = i+1
yield sorted(freqs.iteritems(), key=operator.itemgetter(1), reverse=True)
# #
# The main function # The main function
# #
word_freqs = count_and_sort(sys.argv[1]) for word_freqs in count_and_sort(sys.argv[1]):
for (w, c) in word_freqs[0:25]: print "-----------------------------"
print w, ' - ', c for (w, c) in word_freqs[0:25]:
print w, ' - ', c