2013-05-26 2 views
0

Я использую python 3.3. У меня есть файл csv, но я просто хочу, чтобы последний столбец каждой строки использовался как список. Я могу отобразить это, но я не могу просто сохранить его в виде списка. Вот код, который я использовал.Python создать список из последнего столбца каждой строки в csv

my_list = [] 
with open(home + filePath , newline='') as f: 
    Array = (line.split(',') for line in f.readlines()) 
    for row in Array: 
      #this prints out the whole csv file 
      #this prints out just the last row but I can't use it as a list 
      print(', '.join(row)) 
      print(row[6]) 

    print(my_list) 

Так как я пошел бы о взятии последнего столбца каждой строки (строка [6]) и полагая, что в список, который я мог бы использовать как целые числа?

ответ

2

Используйте csv модуль для простоты использования, то список понимание:

import csv 
import os 

with open(os.path.join(home, filePath), newline='') as f: 
    reader = csv.reader(f) 
    my_list = [row[-1] for row in reader] 

Обратите внимание, что я использую row[-1], чтобы выбрать последний элемент каждой строки.

Ваш код никогда не добавил ничего в my_list; a my_list.append(row[6]) исправил бы это.

+0

Спасибо, что работал отлично. – aldx1

Смежные вопросы