商店的分类/匹配

Question

我想写一个python代码来自动对商店名称(化学家、餐馆等)进行分类。

如果商店的名称是Anand医药商店，它应该落在化学家猫。如果它是7通用商店，它应该落在普通商店猫。

Answer 1

我不知道问题是什么，但如果你在寻找一个关于如何实现它的建议，那么请在下面找到我的想法。

如果您有一个表示不同存储类别的单词列表，则可以创建一个字典，将这些单词作为键，相应的存储类别作为值。

当您想要对商店名称进行分类时，请遍历上面的字典并搜索商店名称中的每个键，当您找到匹配项时，然后将对应于该键的类别分配给商店名称。

Answer 2

您可以尝试使用regexes对以下内容进行分类：

import re
from enum import Enum


class Store(Enum):
    UNKNOWN = 1
    CHEMIST = 2
    RESTAURANT = 3
    GENERAL = 4


CHEMIST_STORE_PATTERN = re.compile(r'.*medical|chemist.*', re.IGNORECASE)
RESTAURANT_STORE_PATTERN = re.compile(r'.*restaurant.*', re.IGNORECASE)
GENERAL_STORE_PATTERN = re.compile(r'.*general.*', re.IGNORECASE)


def classify_store(store: str) -> Store:
    if CHEMIST_STORE_PATTERN.search(store):
        return Store.CHEMIST
    if RESTAURANT_STORE_PATTERN.search(store):
        return Store.RESTAURANT
    if GENERAL_STORE_PATTERN.search(store):
        return Store.GENERAL
    return Store.UNKNOWN


def main() -> None:
    store_to_classify = input('Enter store to classify (for testing only): ')
    print(classify_store(store_to_classify))


if __name__ == '__main__':
    main()

示例用法1:

Enter store to classify (for testing only): Anand Medical store
Store.CHEMIST

示例用法2:

Enter store to classify (for testing only): 7 General store
Store.GENERAL