-1

我有一个包含 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进来,它应该显示这个项目代码。

如果将来出现新关键字,则应将该项目代码添加到字典中。像这样我需要创建一个模型。我需要使用哪种型号...任何人都可以帮助我。

4

1 回答 1

0

无需建立模型。一个简单的字典就足够了。假设您的 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])

并且只需检查配对键是否存在于字典中。如果没有添加它。

于 2017-11-10T06:24:57.103 回答