我是python新手,但是在阅读了3天并找到了解决方案却没有成功之后,我迷失了方向。在
我有mysql表和数据(id,user_id…)。我连接到数据库,读取用户名并将数据保存到数组“user”中。然后我打开csv文件,文件中有很多行和列(用户id、姓名、邮件、电话、地址…)。在下一步中,我将比较db中的user_id与csv文件中的user_id是否匹配。如果答案是“是”,那么我将这一行写入另一个csv文件。在
问题是我的代码只适用于第一个id_用户。拜托,你能帮帮我吗?在
示例表:table in db:
id user_id
1 318604
2 318624
csv文件:
^{pr2}$
新文件中的预期结果:318604; John; john@gmail.com
318604; 053746; USA
318624; Lucy; Lucy@gmail.com
318624; 058839 Sweeden
代码:cur = con.cursor()
with open('input.csv', mode='rb') as f:
reader = csv.reader(f, delimiter=';')
with open('output.csv', mode='a') as w:
writer = csv.writer(w)
with con:
cur.execute("SELECT user_id FROM users")
user=cur.fetchall()
for i in range(len(user)):
for row in reader:
if(user[i][0]==row[0]):
writer.writerow(row)
con.close()