玩蛇网提供最新Python编程技术信息以及Python资源下载!

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为函数失败。
小菜鸟一枚,不知道怎么搞了,求大神指点

可以这样

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()

代码未测试,有问题请简单修改

玩蛇网文章,转载请注明出处和文章网址:https://www.iplaypy.com/wenda/wd13784.html

相关文章 Recommend

玩蛇网Python互助QQ群,欢迎加入-->: 106381465 玩蛇网Python新手群
修订日期:2017年05月23日 - 11时02分36秒 发布自玩蛇网

您现在的位置: 玩蛇网首页 > Python问题解答 > 正文内容
我要分享到:

必知PYTHON教程 Must Know PYTHON Tutorials

必知PYTHON模块 Must Know PYTHON Modules