给定一个段落和一个禁用单词列表。返回出现次数最多,同时不在禁用列表中的单词

给定一个段落 (paragraph) 和一个禁用单词列表 (banned)。返回出现次数最多,同时不在禁用列表中的单词。题目保证至少有一个词不在禁用列表中,而且答案唯一。

禁用列表中的单词用小写字母表示,不含标点符号。段落中的单词不区分大小写。答案都是小写字母。

示例:

输入:
paragraph = "Bob hit a ball, the hit BALL flew far after it was hit."
banned = ["hit"]
输出: "ball"

解释:

“hit” 出现了3次,但它是一个禁用的单词。

“ball” 出现了2次 (同时没有其他单词出现2次),所以它是段落里出现次数最多的,且不在禁用列表中的单词。

注意,所有这些单词在段落里不区分大小写,标点符号需要忽略(即使是紧挨着单词也忽略, 比如 “ball,”),

“hit”不是最终的答案,虽然它出现次数更多,但它在禁用单词列表中。

说明:

1 <= 段落长度 <= 1000.
1 <= 禁用单词个数 <= 100.
1 <= 禁用单词长度 <= 10.
答案是唯一的, 且都是小写字母 (即使在 paragraph 里是大写的,即使是一些特定的名词,答案都是小写的。)
paragraph 只包含字母、空格和下列标点符号!?',;.
paragraph 里单词之间都由空格隔开。
不存在没有连字符或者带有连字符的单词。
单词里只包含字母,不会出现省略号或者其他标点符号。

代码如下:

import unittest

class Solution:
    def mostCommonWord(self, paragraph, banned):
        """
        :type paragraph: str
        :type banned: List[str]
        :rtype: str
        """
        t = paragraph.split(" ")
        word = dict()
        for item in t:
            item = item.lower()
            item = item.strip(",")
            word[item] = word.get(item, 0)+1

        #将禁用词设置为0
        for hit in banned:
            word[hit] = 0

        #取出并排序
        l = [(value, key) for key, value in word.items()]
        l.sort(reverse=True)

        return l[0][1]

class TestmostCommonWordFunc(unittest.TestCase):
    """Test case"""

    def test_mostCommonWord(self):
        """Test method mostCommonWord()"""
        s =Solution()
        paragraph = "Bob hit a ball, the hit BALL flew far after it was hit."
        banned = ["hit"]
        self.assertEqual("ball", s.mostCommonWord(paragraph, banned))

if __name__ == '__main__':
    unittest.main()