在工作中有时需要查询上万行指定的数据,就会用到SQL语句中 select * from table1 where table1.name in (' ' , ' ' ) 的条件查询,所以自己写了个小小的Python脚本来处理这多行数据,废话不多说,上代码:
初级代码:
old_data = open("old_data.txt","r")
new_data = open("new_data.txt","w")
for line in old_data:
line=line.strip('\n')
new_data.write("'"+line+"',")
old_data.close()
new_data.close()
升级代码:
with open('old_data.txt','r') as f1:
with open('new_data.txt','w') as f2:
for line in f1:
line = line.strip('\n')
f2.write("'" + line + "',")
两种代码的效果都是一样的,使用open()函数,最后一步就要手动写调用close()方法,比较麻烦,所以升级后使用with 语句,这样Python就帮我们自动调用close()方法。
效果如下:
不足:处理后的数据应去掉最后一个逗号,这样才是最完整的SQL语句符合where in()条件的数据。