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

我有一个脚本，可以在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")

最佳答案

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

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编写器，然后在您找到文章时写每一行。有关csv编写器的更多信息in the python docs

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

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

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

关于python - Python3:如何将短脚本中的信息存储在CSV文件中？，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/36138838/