我正在编写一个python函数,它使用一个字符串列表并生成一个最常出现的项目列表。
例如:
>>> trending(["banana", "trouble", "StarWars", "StarWars", "banana", "chicken", "BANANA"])
["banana", "StarWars"]
但是
>>> trending(["banana", "trouble", "StarWars", "Starwars", "banana", "chicken"])
["banana"]
到目前为止,我编写的函数只生成频繁出现的第一个单词,而不是频繁出现的单词列表。此外,我的列表包含该频繁项目的索引。
def trending(slst):
words = {}
for word in slst:
if word not in words:
words[word] = 0
words[word] += 1
return words
如何修复此功能以生成最常出现的项目列表(而不是第一个最常出现的项目)以及如何删除索引?
答案 0 :(得分:3)
如果不使用Counter
,您可以使用dict
创建自己的计数器并提取频繁项目:
def trending(slst):
count = {}
items = []
for item in set(slst):
count[item] = slst.count(item)
for k, v in count.items():
if v == max(count.values()):
items.append(k)
return items
答案 1 :(得分:2)
使用Counter
:
In [1]: from collections import Counter
In [2]: l = ["banana", "trouble", "StarWars", "StarWars", "banana", "chicken", "BANANA"]
In [3]: Counter(l)
Out[3]: Counter({'StarWars': 2, 'banana': 2, 'BANANA': 1, 'trouble': 1, 'chicken': 1})
使用Counter(l).most_common(n)
,您可以获得n
最常见的项目。
您的trending()
功能基本上与Counter
的功能相同。计算单词出现次数后,您可以使用max(words.values())
获取最大出现次数。这可用于过滤单词列表:
def trending(slst):
...
max_occ = max(words.values())
return [word for word, occ in words.items() if occ == max_occ]
答案 2 :(得分:0)
以下解决方案仅使用列表。不使用dictionary
,set
或其他Python集合:
def trending(words):
lcounts = [(words.count(word), word) for word in words]
lcounts.sort(reverse=True)
ltrending = []
for count, word in lcounts:
if count == lcounts[0][0]:
if word not in ltrending:
ltrending.append(word)
else:
break
return ltrending
ltests = [
["banana", "trouble", "StarWars", "StarWars", "banana", "chicken", "BANANA"],
["banana", "trouble", "StarWars", "Starwars", "banana", "chicken"]]
for test in ltests:
print trending(test)
它提供以下输出:
['banana', 'StarWars']
['banana']