Python 帮助 - 在 Python 中执行“isnumber(search”)

Question

我一直在寻找这个答案的高低，我在任何地方都找不到合适的解决方案。

我想对列中的一组关键字进行查看和分类，我通常使用“isnumber(search”) 执行此操作，但我限制为 64 个变体。我使用的代码示例如下：

=IF(ISNUMBER(SEARCH("bag",A2)),"Bags",
IF(ISNUMBER(SEARCH("rucksack",A2)),"Bags",
IF(ISNUMBER(SEARCH("handbag",A2)),"Bags",
IF(ISNUMBER(SEARCH("purse",A2)),"Personal Bags",
IF(ISNUMBER(SEARCH("wallet",A2)),"Personal Bags")))))

预期的输出如下：

Keyword	Category
handbag	Bags
rucksack	Bags
drawstring bags	Bags
tote bag	Bags
cotton tote	FALSE
purse	Personal Bags
wallet	Personal Bags

可以在 Python 中复制相同的功能吗？

我希望布局为 A 列

我见过这种变体，但它使用“真”或“假”语句，并在“关键字”列的右侧添加了无穷无尽的列，但清理起来很费力。

如果有人知道如何在 Python 中复制 Excel 公式，将不胜感激 :)

干杯

Answer 1

您想搜索项目集以查看任何特定项目属于哪个集？

def categoriseItems(currVar):
    bags = set(["handbag", "rucksack", "drawstring bags", "tote bag"])
    personalBags = set(["purse", "wallet"])

    itemCat = None
    if currVar in bags:
        itemCat = "bags"
    elif currVar in personalBags:
        itemCat = "personal bags"

    if not itemCat:
        print("Item " + currVar + " not categorised")

    return [currVar, itemCat]

编辑：

那么，如果你想要你的输出格式，是 openpyxl 吗？

如果是这样，将数据组合成一个列表，如下所示：

rows = [
       ["Keyword",  "Category"],
       ]

然后，将我上面的内容作为每个项目的内联 for 循环或作为每个项目调用的单独函数（重新排列为单独的函数）：

currVar = A2.text   #If your lifting this from a spreadsheet, openpyxl is prob your goto library - this line is just pseudocode

下面假设您有一个与您要处理的项目相对应的文本条目列表：

for currVar in listOfVars:
    rows.append(categoriseItem(currVar))



def writeToWB(rows):
    import openpyxl
    wb = openpyxl.Workbook()
    ws = wb.active

    for row in rows:
        ws.append(row)

    wb.save("myWorkbook.xlsx")

抱歉，有点脱节 - 我赶时间！

Answer 2

假设您想对数据进行桶排序/分箱/分类，您可以使用 python dict 进行映射：

keyword = A2.text
mapping = {'handbag': 'Bags', 'rucksack': 'Bags', 'cotton tote': false}
try:
    result = mapping[keyword]
except KeyError:
    print("Keyword not found")

ws['B2'] = result

如果您想避免 if-chain，这也可以与 Amiga500 的答案结合使用。