Czytanie plików .csv na listach Pythona

Question

Mam dużo plików .csv w katalogu i chciałbym otworzyć każdy z nich w pętli w Pythonie tak, aby pierwszy .csv został odczytany na liście [0] i drugi .csv jest czytany na liście [1] i tak dalej.

Niestety, podczas gdy mój kod pętli przez wszystkie pliki .csv, umieszcza wszystkie pliki .csv na liście [0]. Jak mogę zmodyfikować mój kod, aby osiągnąć powyższy cel? Wielkie dzięki.

John

Oto kod:

def create_data_lists(): 
i=0 
for symbol in symbols: 
    with open(symbols[i]+'.csv', 'r') as f: 
     print i 
     reader = csv.reader(f) 
     reader.next() 
     for row in reader: 
      rowdata.append(row) 
    data_by_symbol.append(rowdata) 
    i=i+1 

Answer 1

wewnątrz pętli for, w pobliżu szczytu, trzeba odświeżyć listę rowdata. w przeciwnym razie dodajesz do tego na zawsze. mieć coś podobnego rowdata = [] zaraz po print i

def create_data_lists(): 

    for symbol in symbols: 
     with open(symbol+'.csv', 'r') as f: 
     print symbol 
     rowdata = [] 
     reader = csv.reader(f) 
     reader.next() 
     for row in reader: 
      rowdata.append(row) 
    data_by_symbol.append(rowdata) 

EDIT pozbył i, jak ja nie bardzo go za pomocą

Answer 2

dlaczego nie przechowuje samych czytelników w liście?

list_of_csv_files = [] 

for f in filenames: 
    list_of_csv_files.append(csv.DictReader(open(f))) 

Spowoduje to zapisanie samego czytelnika na liście, co pozwala później na coś takiego jak:

for row in list_of_csv_files[0]: 
    # do some processing on the row 

Największą zaletą tej metody jest taht następnie można zrobić rzeczy jak kolumnach filtracyjnych łatwo, stosując metody takie jak:

one_row = [row["name of column heading"] for row in list_of_csv_files[0]] 
two_rows = [[row["name col 2"], row["name col 2"]] for row in list_of_csv_files[0]] 

który podejrzewam byłoby bardziej pomocne niż przechowywanie programu pre-odczytu (a więc de-uporządkowany) plików CSV.

ale jeśli naprawdę chcesz mieć wszystkie pliki CSV odczytane i zapisane na liście, trzeba będzie wykaz list, ja nie polecam, to będzie bardzo intensywny pamięci:

list_of_csv_files = [[]] 

for f in filenames: 
    list_of_csv_files.append([row.values() for row in csv.DictReader(open(f))])