用 Python 实现函数 count_words(),该函数输入字符串 s 和数字 n,返回 s 中 n 个出现频率最高的单词。返回值是一个元组列表,包含出现次数最高的 n 个单词及其次数,即 [(<单词1>, <次数1>), (<单词2>, <次数2>), ... ],按出现次数降序排列。
"""Count words."""
def count_words(s, n):
"""Return the n most frequently occuring words in s."""
# TODO: Count the number of occurences of each word in s
list_tmp = s.split()
list_split = list(set(list_tmp))
list_ans = []
for item in list_split:
list_ans.append((item,list_tmp.count(item)))
# TODO: Sort the occurences in descending order (alphabetically in case of ties)
list_ans = sorted(list_ans, key=lambda x:(-x[1],x[0]))
# TODO: Return the top n most frequent words.
top_n = []
for i in range(n):
top_n.append(list_ans[i])
return top_n
def test_run():
"""Test count_words() with some inputs."""
print count_words("cat bat mat cat bat cat", 3)
print count_words("betty bought a bit of butter but the butter was bitter", 3)
if __name__ == '__main__':
test_run()