循环从csv检索数据,附加到文件

时间:2015-05-22 00:22:07

标签: python python-2.7 loops csv

我创建了一个Python 2.7脚本,它执行以下操作:

  
      
  1. 从文件夹中获取文件名列表,并将其写入csv文件,每行一个。
  2.   

并且

  
      
  1. 将数据输入网络搜索框。
  2.   
  3. 将搜索框中的结果写入另一个csv文件。
  4.   

所以我现在想要的是(1)中的csv数据作为(2)的输入。

即。对于csv文件中的每个文件名,它会搜索该单元格。

此外,我不想将结果写入(3)中的第二个csv文件,而是将结果附加到第一个csv文件中 - 或者生成一个包含两个列的新文件。

我可以提供代码,但由于它已经有50行,我只是试着让这个问题具有描述性。

更新:建议检索和追加:

with open("file.csv","a+") as f:
    r = csv.reader(f)
    wr = csv.writer(f, delimiter="\n")
    result = []
    for line in r:
        searchbox = driver.find_element_by_name("searchbox")
        searchbox.send_keys(line)

        sleep(8)

        search_reply = driver.find_element_by_class_name("search_reply")

        result = re.findall("((?<=\()[0-9]*)", search_reply.text)
    wr.writerow(result)

1 个答案:

答案 0 :(得分:1)

打开阅读和追加,存储输出然后写在最后:

import csv
with open("first.csv","a+") as f:
    r = csv.reader(f)
    wr = csv.writer(f,delimiter="\n")
    result = []
    for line in r:
        # process lines/step 2
        # append to result
    wr.writerow(result)