Добавление отдельных строк из csv в python

Question

Я хочу добавить отдельные строки из файла csv. Я хочу добавить первый, третий, четвертый и пятый элементы. Вот что я до сих пор, но, похоже, это не работает - какие-то идеи? Этот код в настоящее время сохраняется в двух отдельных списках; Я хочу, чтобы строки были добавлены в один список.

import csv 
game_file = open(gamefile + ".csv") 
game_file = [] 
score_file.append([row[0]]) 
score_file.append([row[2:5]]) 
with open(studentsclass + ".csv", "w") as f: 
     writer = csv.writer(f) 
     writer.writerows(score_file) 
     f.close() 
    game_file.close() 

Answer 1

Использование csv может помочь:

import csv 

game_file = [] 
included_cols = [0, 2, 3, 4, 5] 

with open(filename, 'rb') as csvfile: 
    reader = csv.reader(csvfile, delimiter=';', quotechar='"') 
    for row in reader: 
     game_file.append(list(row[i] for i in included_cols)) 

Answer 2

Вы можете использовать перечисление на вашем reader объекта и проверить, если индекс в желаемом списке (который я преобразовал его, чтобы установить, что имеет O (1) для проверки членства):

import csv 

temp=0 
rows=[] 
with open('eggs.csv', 'rb') as csvfile: 
    spamreader = csv.reader(csvfile, delimiter=' ', quotechar='|') 
    for i,row in enumerate(spamreader): 
      if i in {0,2,3,4,5}: 
       temp+=1 
       rows.append(row) 
      if temp==5: 
        print rows 
        break