您的位置:首页 > 编程语言 > Python开发

【Python系列3】两个txt文件通过id进行关联

2015-09-04 22:32 826 查看
       进行不同表的关联,我们首先想到的都是通过SQL在数据库中完成。然而,最近在做项目的过程中,却遇到了两个表都保存为txt文件,并且需要按照ID进行关联的问题。为了解决这个问题,自己写了下面一段Python程序,主要思想是将一个txt中的ID作为字典的key。

def JointTWOTXT(self): #将A.txt和B.txt按照ID进行关联
f=open('A.txt','r')
ff=open('B.txt','r')
fff=open('C.txt','w') # 关联之后,字段1:ID,字段2:NAME,字段3:Menu,字段4:category_id

dicc=dict() # 将A.txt转为字典,key为id,value为category_id
for line in ff:
line=line.strip().split('|')
# print line[0]
# print line[1]
dicc[line[0]]=line[1]

id_list=[] # ID列表
for line1 in f:
ID1=line1.split()[0]
# print ID1
NAME=line1.split()[1]
# print NAME
Menu=line1.split()[2]
# print Menu
if (dicc.get(str(ID1))!=None): # ID1在字典中
if (ID1 not in id_list): # 去掉重复
id_list.append(ID1)
category_id=dicc[str(ID1)]
fff.write(str(ID1)+str('|||')+str(NAME)+str('|||')+str(Menu)+str('|||')+str(category_id))
fff.write(str('|||')+'\n')
fff.close()
ff.close()
f.close()
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  python txt关联