首頁  >  問答  >  主體

python - for循环print怎样才能输出csv呢

import csv,re

def search(req,line):
    text = re.search(req,line)
    if text:
        data = text.group(1)
    else:
        data = 'no'
    return data

csvfile = file('serp_html.csv','rb')
reader = csv.reader(csvfile)

'''输出百度搜索结果数据:当前关键词,排名,排名网站,百度url(需转义后才是真实的url),标题'''
for line in reader:
    word = line[0]
    html = line[1]

    number = search(r'id="(\d+)"',html)
    domain = search(r'<span class="g">(.*?)/.*</span>',html)
    bdurl = search(r'href="(http://www.baidu.com/link\?url=[^"]*?)"',html)
    title = search(r'"title":"([^"]*?)"',html)

    print '%s,%s,%s,%s,%s' % (word,number,domain,bdurl,title)

以上是一个继承程序,运行后能print出正确结果,但是我希望能生成csv报表文件,尝试修改for为函数失败。
小菜鸟一枚,不知道怎么搞了,求大神指点

大家讲道理大家讲道理2740 天前725

全部回覆(1)我來回復

  • 黄舟

    黄舟2017-04-18 10:26:57

    可以這樣

    import csv,re
    
    def search(req,line):
        text = re.search(req,line)
        if text:
            data = text.group(1)
        else:
            data = 'no'
        return data
    
    reuslts = []
    result_csv = file('new_file.csv', 'wb')
    result_csv_writer = csv.writer(result_csv)
    '''输出百度搜索结果数据:当前关键词,排名,排名网站,百度url(需转义后才是真实的url),标题'''
    # 保存标题
    result_csv_writer.writerow(['关键词', '排名', '排名网站', '百度url', '标题']) 
    for line in reader:
        word = line[0]
        html = line[1]
    
        number = search(r'id="(\d+)"',html)
        domain = search(r'<span class="g">(.*?)/.*</span>',html)
        bdurl = search(r'href="(http://www.baidu.com/link\?url=[^"]*?)"',html)
        title = search(r'"title":"([^"]*?)"',html)
    
        reuslts.append((word, number, domain, bdurl, title))
        # print '%s,%s,%s,%s,%s' % (word,number,domain,bdurl,title)
    
    # 保存多行
    result_csv_writer.writerows(reuslts)
    result_csv.close()

    程式碼未測試,有問題請簡單修改

    回覆
    0
  • 取消回覆