1773. 统计匹配检索规则的物品数量 python

1773. 统计匹配检索规则的物品数量

题目链接
示例 1：

输入：items = [[“phone”,“blue”,“pixel”],[“computer”,“silver”,“lenovo”],[“phone”,“gold”,“iphone”]], ruleKey = “color”, ruleValue = “silver”
输出：1

解释：只有一件物品匹配检索规则，这件物品是 [“computer”,“silver”,“lenovo”] 。

示例 2：

输入：items = [[“phone”,“blue”,“pixel”],[“computer”,“silver”,“phone”],[“phone”,“gold”,“iphone”]], ruleKey = “type”, ruleValue = “phone”
输出：2

解释：只有两件物品匹配检索规则，这两件物品分别是 [“phone”,“blue”,“pixel”] 和 [“phone”,“gold”,“iphone”] 。注意，[“computer”,“silver”,“phone”] 未匹配检索规则。

代码：

class Solution(object):
    def countMatches(self, items, ruleKey, ruleValue):
        """
        :type items: List[List[str]]
        :type ruleKey: str
        :type ruleValue: str
        :rtype: int
        """
        summ = 0
        col = 0
        if ruleKey == "type":
            col = 0
        elif ruleKey == "color":
            col = 1
        else:
            col = 2
        for item in items:
            if item[col] == ruleValue:
                summ += 1
        return summ

提交结果：