Исправление IndexError: список проблем вне диапазона

Question

Я работаю с файлом .csv, я написал этот код, чтобы подсчитать, сколько раз каждое значение в столбце year встречается в наборе данных csv.

Я получаю IndexError: list out of range в строке 10 на row_year = suspension[5] всякий раз, когда я запускаю код в своей персональной системе, но код работает нормально, когда я запускаю его на сайте dataquest.

Набор данных CSV имеет 7 столбцов, 5-й столбец представляет лет .

import csv

file  = open("nfl_suspensions_data.csv")
nfl_suspensions = list(csv.reader(file))
nfl_suspensions = nfl_suspensions[1:]

years = {}

for suspension in nfl_suspensions:
    row_year = suspension[5]
    if row_year in years:
        years[row_year] = years[row_year] + 1
    else:
        years[row_year] = 1

print(years)

Answer 1

У вас слишком короткие данные - вы попадаете в список позади них. Если у вас действительно есть year в качестве 5-го столбца, вы должны использовать column[4] для доступа к нему - индексы python основаны на 0.

Использовать Обработка ошибок :

import csv

file  = open("nfl_suspensions_data.csv")
nfl_suspensions = list(csv.reader(file))
nfl_suspensions = nfl_suspensions[1:]

years = {}

 for line_nr, suspension in enumerate(nfl_suspensions):
    try:
        row_year = suspension[5]
    except IndexError:
        # 0 based line_nr, line_nr + 1 due to removed header line  
        print("Data corrupt: less then 6 entries. Line:", line_nr+1)
        print(suspension)

        # skip this data
        continue

    if row_year in years:
        years[row_year] = years[row_year] + 1
    else:
        years[row_year] = 1

print(years)

Это следует за питонами Просить прощения, а не разрешения Философия.

---

Вы должны также переключиться на

with open("nfl_suspensions_data.csv") as file:
    nfl_suspensions = list(csv.reader(file))[1:]

которая является предпочтительной ва чтения файлов. См. python.org - чтение и запись файлов (см. 2-й пример кода)

Вы также можете использовать collections.defaultdict:

years = defaultdict(int) # above 

и удалите проверки if вокруг

# if row_year in years:
    years[row_year] += 1  # this should work using a defaultdict(int)
# else:
#    years[row_year] = 1

или используйте collections.Counter

---

Более короткий код, включая генерацию файла (с указанием года в строке [5] == 6-й столбец), который выполняет вашу задачу:

import csv
from collections import Counter


# Create a demo data file with errors:    
with open("nfl_suspensions_data.csv","w") as f:
    for inter in range(1,10):
        for y in range(1980,2001,inter):
            f.write(f"na,na,na,na,na,{y},na,na\n")
        # corrupt line
        f.write(f"na,na,na,na\n")


# process and count the years:
with open("nfl_suspensions_data.csv") as file:
    nfl_suspensions = list(csv.reader(file))[1:]

as_columns = list(zip(*[l for l in nfl_suspensions if len(l) > 6]))
print(Counter(as_columns[5]))

Выход:

Counter({'1980': 8, '1992': 5, '1998': 5, '1986': 4, '1988': 4, '1996': 4,
         '2000': 4, '1984': 3, '1989': 3, '1990': 3, '1994': 3, '1995': 3, 
         '1982': 2, '1983': 2, '1985': 2, '1987': 2, '1981': 1, '1991': 1, 
         '1993': 1, '1997': 1, '1999': 1})

---

Ваша логика исправлена, применена к сгенерированным выше данным:

def your_code_fixed(sus):
    years = {}
    for line_nr, suspension in enumerate(sus):
        try:
            row_year = suspension[5]
        except IndexError:
            # 0 based line_nr, line_nr + 1 due to removed header line  
            print("Data corrupt: less then 6 entries. Line:", line_nr+1)
            print(suspension)

            # skip this data
            continue
        if row_year in years:
            years[row_year] = years[row_year] + 1
        else:
            years[row_year] = 1
    print(years)    

with open("nfl_suspensions_data.csv") as file:
    nfl_suspensions = list(csv.reader(file))[1:]

your_code_fixed(nfl_suspensions)

Вывод с указанным файлом данных:

Data corrupt: less then 6 entries. Line: 21
['na', 'na', 'na', 'na']
Data corrupt: less then 6 entries. Line: 33
['na', 'na', 'na', 'na']
Data corrupt: less then 6 entries. Line: 41
['na', 'na', 'na', 'na']
Data corrupt: less then 6 entries. Line: 48
['na', 'na', 'na', 'na']
Data corrupt: less then 6 entries. Line: 54
['na', 'na', 'na', 'na']
Data corrupt: less then 6 entries. Line: 59
['na', 'na', 'na', 'na']
Data corrupt: less then 6 entries. Line: 63
['na', 'na', 'na', 'na']
Data corrupt: less then 6 entries. Line: 67
['na', 'na', 'na', 'na']
Data corrupt: less then 6 entries. Line: 71
['na', 'na', 'na', 'na']

{'1981': 1, '1982': 2, '1983': 2, '1984': 3, '1985': 2, '1986': 4, '1987': 2,
 '1988': 4, '1989': 3, '1990': 3, '1991': 1, '1992': 5, '1993': 1, '1994': 3, 
 '1995': 3, '1996': 4, '1997': 1, '1998': 5, '1999': 1, '2000': 4, '1980': 8}