Как импортировать определенный ключ со своими значениями в CSV-файл, используя Python? - PullRequest
Новая
       16

Как импортировать определенный ключ со своими значениями в CSV-файл, используя Python?

0 голосов
/ 17 мая 2019

В моей программе пользователь будет вводить название страны, и мой код будет первым, независимо от того, существует ли название страны в словаре.Если он существует, код должен напечатать название страны и связанные с ней значения в CSV-файле.Тем не менее, я не могу сделать часть CSV.Я представил словарь и соответствующий код для этой проблемы.Пожалуйста, дайте мне знать, в чем проблема. 

import string
import re
import csv

data = {
    "Pakistan": (0.57, 0.05, 0.79),
    "India": (0.47, 0.12, 0.54),
    "Bangladesh": (0.49, 0.17, 0.81)
    }
csv_columns = ['Country Name','1997','1998','1999','2000','2001','2002','2003','2004','2005','2006','2007','2008','2009','2010']
csv_file = "Emissions_subset.csv"

con_name = input("Write up to three comma-separated countries for which you want to extract data: ")
count = len(re.findall(r'\w+', con_name))
if count == 1:
    con_check1 = con_name.split()[0]
    if con_check1.lower() in map(str.lower, data.keys()):
        with open(csv_file, 'w') as csvfile:
            writer = csv.DictWriter(csvfile, fieldnames=csv_columns)
            writer.writeheader()
            writer.writerow(list(data.keys()).index(con_check1))
        print("Data successfully extracted for countries {} saved into file Emissions_subset.csv".format(con_check1.capitalize()))
...