Python3：如何将短脚本中的信息存储在CSV文件中？

Question

我有一个脚本，可以在Google上搜索Lil Wayne的文章，然后-对于每篇文章-返回标题，摘要，URL和关键字。

但是我真的很想制作一个以TITLE，SUMMARY，URL，KEYWORDS为列的CSV文件，然后在每一行中存储每篇文章的相关信息。

from newspaper import Article 
import google 

#Search Setup
for url in google.search('Lil Wayne', num=10, stop=3, pause=0):
    article = Article(url)      
    article.download() 
    article.parse()
    article.nlp()     


    #Print the parsed output of each article
    print(u'TITLE: ' + str(article.title.encode('ascii', 'ignore')))
    print(u'SUMMARY: ' + str(article.summary.encode('ascii', 'ignore')))
    print(u'URL: ' + str(article.url))
    print(u'KEYWORDS: ' + str(article.keywords)) 
    print("\n")

Answer 1

您可以在代码中使用如下代码：

from newspaper import Article 
import google 

with open('output_file.csv', 'wb') as csvfile:
    lil_wayne_writer = csv.writer(csvfile)

    #Search Setup
    for url in google.search('Lil Wayne', num=10, stop=3, pause=0):
        article = Article(url)      
        article.download() 
        article.parse()
        article.nlp()
        lil_wayne_writer.writerow(
            [
                str(article.title.encode('ascii', 'ignore')),
                str(article.summary.encode('ascii', 'ignore')),
                str(article.url),
                str(article.keywords),
            ]
        )

基本上，这将打开一个csv编写器，然后在找到文章时写入每一行。 在python文档中有关于csv编写器的更多信息

您可能需要对其进行一些编辑才能在您的环境中正常工作。

如果您想将标头写入CSV文件，只需将一个调用添加到以下内容：

lil_wayne_writer.writerow(['TITLE', 'SUMMARY', 'URL', 'KEYWORDS'])