Печать элементов из файла csv с использованием raw_input

Question

Как печатать элементы из файла csv с использованием raw_input?

Код:

import csv 

inputYear = raw_input('Please enter the year: ') 
inFile = open('babyQldAll.csv', 'rU') 
cvsFile = csv.reader(inFile, delimiter=',') 

dict = {} 

for row in cvsFile: 
    year, name, count, gender = row 

    if (year == inputYear) and (gender == 'Boy'): 
     dict[name] = count 

     print('Popular boy names in year %s are:' % inputYear) 

# According to informaiton in 'dict', print (name, count) sorted by 'name' in alphabetical order   

print("Print boy names... ")  
inFile.close() 

Желаемый результат:

Популярные имена мальчика в 2010 году являются: Айдан 112 Эйден 168 Alex 93

Answer 1

Есть много возможных способов сделать это.
Один из них - построить диктофон, где годами являются ключи, а значения - список списков, например [[John, 124, Male], [Amy, 111, Female]] i.e пары, указывающие имя, счет и пол. Вы можете создать полный словарь, а затем получить значения для каждого года:

import csv 

inputYear = raw_input('Please enter the year: ') 
inFile = open('babyQldAll.csv', 'rU') 
cvsFile = csv.reader(inFile, delimiter=',') 

dict = {} 

for row in cvsFile: 
    year, name, count, gender = row 

    if dict.get(year, None): #If key in dict already 
     dict[year].append([name, count, gender]) 
    else: #else, create the key 
     dict[year] = [[name, count, gender]] 

data = ", ".join(" ".join(z) for z in dict.get(inputYear,[]) if z[3]=='Male') 
if len(data): 
    print 'Popular boy names in {} are: {}'.format(inputYear, data) 

inFile.close() 

Answer 2

Вы можете попробовать это: (?)

inputYear = raw_input('Please enter the year: ') 

file1 = open('babyQldAll.csv','r') 
fileContent = file1.readlines() 
fileRows = fileContent[0].split("\\n") 

dict = {} 
for row in fileRows: 
    year = row.split(',')[0].strip() 
    name = row.split(',')[1].strip() 
    count = row.split(',')[2].strip() 
    gender = row.split(',')[3].strip() 
    if (year == inputYear) and (gender == 'Boy'): 
     dict[name] = count 
odDict = sorted(dict.items()) 
str = '' 
for i in odDict: 
    str = str + i[0] + ' ' + i[1] + ' ' 

print('Popular boy names in year '+ inputYear + ' are: ' + str) 
file1.close()