將Biopython輸出寫入CSV

Question

我目前正在一個班級項目上，需要我使用biopython從NCBI網站提取數據並將其寫入CSV文件，然后在R中進行分析。我已獲得所需的所有數據，但我不確定如何將其寫入CSV文件，因為我們從未在課堂上介紹過它。 到目前為止，這是我的代碼：

from Bio import Entrez, Medline

Entrez.email = "email.here"

handle = Entrez.esearch(db="pubmed",  # database to search
                        term="Chan CS[Author] AND 2000:2017[Date - Publication]",  # search term
                        retmax=200 # Maximum number of results to return
                        )
record = Entrez.read(handle)
handle.close()

pmid_list = record["IdList"]
print(pmid_list)

其次是

from Bio import Medline
handle = Entrez.efetch(db="pubmed", id=pmid_list, rettype="medline", retmode="text")
records = Medline.parse(handle)

journal_dict = []
datep_dict = []
place_dict = []
for record in records:

    # retrieve journal titles 
    title = record['JT']
    journal_dict.append(title)

    #retrieve date published
    date = record['DP']
    datep_dict.append(date)

    #retrieve place published
    place = record['PL']
    place_dict.append(place)
# Close the efetch handle    
handle.close()

for title in journal_dict:
    print(title)
for date in datep_dict:
    print(date)
for place in place_dict:
    print(place)

最后，我堅持的部分

import csv

我正在嘗試使csv文件看起來像下面的樣子

[ID, Journal Title, Publication Date, Place of Publication]
[123, Title1, Date1, Place1]
[124, Title2, Date2, Place2]

任何幫助將不勝感激！

Answer 1

在第二段代碼中，您的變量名談論dict ，但實際上它們是list s：

journal_dict = []
datep_dict = []
place_dict = []

因此，讓我們對其進行修復（稍后在寫入CSV時也將有用）：

record_list = []
for record in records:
    record_dict = {'ID': record['ID'],
                   'Journal Title': record['JT'],
                   'Publication Date': record['DP']
                   'Place of Publication': record['PL']}
    record_list.append(record_dict)

現在，讓我們將此字典列表寫入CSV文件

import csv

with open('medline.csv', 'w', newline='') as csvfile:
    fieldnames = ['ID', 'Journal Title', 'Publication Date', 'Place of Publication']
    writer = csv.DictWriter(csvfile, fieldnames=fieldnames)

    writer.writeheader()
    for record_dict in record_list:
        writer.writerow(record_dict)