Python:模型构建 - 选择哪种模型

Question

我有一个包含 4 列的数据框，我的数据库如下所示：

Sl.No Column-1 Column-2 Item code 1 gaddhjhd,dfhhaso dfkjdhf 3027-121 2 fdlkdsa,fdhjudf dfhjdiufu 3027-320 3 dfhidfndf eioruoier 3027-343 4 rieruiq jdjlasjd 3027-420 5 lkjdfi dfjdflajdfi 3027-720

现在我需要用python构建一个模型，机器必须读取column-1 & Column-2并创建一个字典，如果将来Column-1 & Column-2中的关键字出现，它应该显示此项目代码。

如果将来有新关键字出现，它应该将该项目代码添加到字典中。像这样，我需要创建一个模型。我需要使用哪种型号...谁能帮我解决这个问题。

Answer 1

无需构建模型。只需一本简单的字典就足够了。假设您的 csv 文件名为"abc.csv"示例代码：

import csv
f = open('./abc.csv')
csv_f = csv.reader(f, delimiter=',')
next(csv_f) #for header skip
model_dict = dict()
for row in csv_f:
    col_1 = str(row[1])
    col_2 = str(row[2])
    model_dict[(col_1,col_2)] = str(row[3])

只需检查配对键是否存在于字典中即可。如果没有，请添加它。