問題描述

import csv,redef search(req,line): text = re.search(req,line) if text:data = text.group(1) else:data = ’no’ return datacsvfile = file(’serp_html.csv’,’rb’)reader = csv.reader(csvfile)’’’輸出百度搜索結(jié)果數(shù)據(jù)：當前關(guān)鍵詞，排名，排名網(wǎng)站，百度url（需轉(zhuǎn)義后才是真實的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://m.b3g6.com/wenda/(http://www.baidu.com/link?url=[^']*?)'’,html) title = search(r’'title':'([^']*?)'’,html) print ’%s,%s,%s,%s,%s’ % (word,number,domain,bdurl,title)

以上是一個繼承程序，運行后能print出正確結(jié)果，但是我希望能生成csv報表文件，嘗試修改for為函數(shù)失敗。小菜鳥一枚，不知道怎么搞了，求大神指點

問題解答

可以這樣

import csv,redef search(req,line): text = re.search(req,line) if text:data = text.group(1) else:data = ’no’ return datareuslts = []result_csv = file(’new_file.csv’, ’wb’)result_csv_writer = csv.writer(result_csv)’’’輸出百度搜索結(jié)果數(shù)據(jù)：當前關(guān)鍵詞，排名，排名網(wǎng)站，百度url（需轉(zhuǎn)義后才是真實的url），標題’’’# 保存標題result_csv_writer.writerow([’關(guān)鍵詞’, ’排名’, ’排名網(wǎng)站’, ’百度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://m.b3g6.com/wenda/(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()

代碼未測試，有問題請簡單修改