2017-10-20 62 views
-1

和我的数据库看起来像下面:的Python:建模 - 哪些型号选择我有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中的关键字出现,它应显示此项目代码。

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

回答

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]) 

而只是检查对关键,如果它存在于词典或没有。 如果没有添加它。

相关问题