Chương trình đếm số mục khớp quy tắc bằng Python

Cho mảng items, mỗi phần tử [type_i, color_i, name_i]. Cho ruleKey ("type"/"color"/"name") và ruleValue. Đếm số item khớp: ruleKey="type" và ruleValue=type_i, hoặc tương tự cho color, name.

Ví dụ: items = [["Bike","blue","ElecB"],["Car","silver","Sumo"],["Bike","blue","TVS"]], ruleKey="color", ruleValue="blue" → 2 (hai Bike xanh).

Thuật toán giải quyết

• count := 0.
• Xác định chỉ số c: 0 nếu type, 1 nếu color, 2 nếu name.
• Duyệt items: nếu item[c] == ruleValue: count++.
• Trả về count.

Ví dụ triển khai

def solve(items, ruleKey, ruleValue):
    count = 0
    key_to_idx = {"type": 0, "color": 1, "name": 2}
    idx = key_to_idx[ruleKey]
    for item in items:
        if item[idx] == ruleValue:
            count += 1
    return count

items = [["Bike","blue","ElecB"],["Car","silver","Sumo"],["Bike","blue","TVS"]]
ruleKey = "color"
ruleValue = "blue"
print(solve(items, ruleKey, ruleValue))

Đầu vào

[["Bike","blue","ElecB"],["Car","silver","Sumo"],["Bike","blue","TVS"]], "color", "blue"

Đầu ra

2