问 题
下面是题目要求:
用 Python 实现函数 count_words(),该函数输入字符串 s 和数字 n,返回 s 中 n 个出现频率最高的单词。返回值是一个元组列表,包含出现次数最高的 n 个单词及其次数,即 [(, ), (, ), ... ],按出现次数降序排列。
您可以假设所有输入都是小写形式,并且不含标点符号或其他字符(只包含字母和单个空格)。如果出现次数相同,则按字母顺序排列。
例如:
print count_words("betty bought a bit of butter but the butter was bitter",3)
输出:
[('butter', 2), ('a', 1), ('betty', 1)]
我已经按照单词出现频率排好了顺序,但是怎么在这基础上再实现字母排序?
解决方案
這應該滿足你的條件,只是我自認為寫得很醜,等有時間再優化吧:
(改了一下 code 這樣做比較好!)
import collections
def count_words(s, n):
lst = collections.Counter(s.split()).most_common()
lst.sort(key=lambda t: t[0])
lst.sort(key=lambda t: t[1], reverse=True)
return lst[0:n]
print(count_words("betty bought a bit of butter but the butter was bitter", 3))
想了一想,覺得下面更好:
import collections
def count_words(s, n):
lst = collections.Counter(s.split()).most_common()
lst.sort(key=lambda t: (-t[1], t[0]))
return lst[0:n]
print(count_words("betty bought a bit of butter but the butter was bitter", 3))
扫一扫关注IT屋
微信公众号搜索 “ IT屋 ” ,选择关注与百万开发者在一起