从列表中查找单元格值,然后将结果写入正确的列

2024-10-01 15:49:37 发布

您现在位置:Python中文网/ 问答频道 /正文

我有一个产品清单,其中包括约9000个产品,以及这个我有约60个产品组。每个产品都是一个或多个产品组的成员,其中一些产品没有任何组成员资格。你知道吗

如果我运行以下代码

PRODUCTGROUPS = {'PRODUCTGROUP-A': ['product1', 'product3'], 'PRODUCTGROUP-B': ['product3'], 'PRODUCTGROUP-C': []}

HEADER = ['Products'] + [*PRODUCTGROUPS]

with open('products.csv', 'w', newline='') as csvfile:
    groupwriter = csv.writer(csvfile, delimiter=',',
                            quotechar='"', quoting=csv.QUOTE_MINIMAL)

    groupwriter.writerow(HEADER)

    for product in products:
        print(product)
        groupwriter.writerow([product])

它给我以下csv文件。(PRODUCTGROUPS是一个嵌套字典,其中包含组及其成员(产品)

Products       | PRODUCTGROUP-A | PRODUCTGROUP-B | PRODUCTGROUP-C
product1
product2
product3

我想在PRODUCTGROUPS中查找产品名称,如果它存在于产品组中,我应该标记相应的单元格。像这样的事

Products       | PRODUCTGROUP-A | PRODUCTGROUP-B | PRODUCTGROUP-C
product1       |      X         |                |
product2       |                |                |
product3       |      X         |        X       |

我可以使用列表理解来检查组成员身份,但是我找不到标记相应单元格的方法。你知道吗


Tags: csvcsvfile产品成员productproductsheaderproduct1

热门问题