Выходит ошибка key error, но график строится и ключ есть. Как исправить?

Question

[basilevis]

plt.figure(figsize=(12,6))
plt.title('Количество продаж популярных игровых платформ')
plt.xlabel('Годы игровых релизов')
plt.ylabel('Продажи')

for i in list(top_5_platforms['platform']):
    sns.lineplot(data=year_total_sale_for_platform(i, df)['total_sales'], label=i)
    plt.legend()

Выводит ошибку KeyError: 'total_sales', но график тоже выходит и такой, как надо. С таблицей, к которой обращаемся, все ок. Пробовала тип данных столбца менять, все равно ошибка.
В чем проблема и как убрать ошибку?

Answer 1

[OperLop]

try:
except KeyError:

Это пробовали? Это код для обхода ошибок. Если у вас он работает и вам нужно избежать ошибки - это решение

Comments

[shurshur]

Вот самое неправильное что надо делать с ошибкой - это заметать её под ковёр. А вдруг в итоге через год обнаружится, что ошибка-то означала неправильность программы вообще и считала она погоду на Марсе?

[OperLop]

Если человек точно уверен в том, что программа явно выдает правильный ответ, то это не важно. А если он действительно не знал о том, что есть возможность обходить ошибку, то наверное это будет очень полезно, ибо у вас, как и у многих наверняка есть программы, где без обхода не обойтись.

[shurshur]

OperLop, если я ловлю exception, то конкретный, и как правило делаю что-то связанное именно с ним. Например, у меня тут есть код, который читает бинарный файл, и если прочитано меньше байт, чем ожидалось (к сожалению, в питоне нет нормального способа получить EOF), то кидает кастомное исключение, по которому вызывающая функция прерывает итерацию цикла.

Ловить любые ошибки без разбору и даже не выводить их пользователю уж точно не надо привыкать.

[o5a]

OperLop, basilevis,
Правильнее было бы в случае ошибки вывести те данные, при которых она возникает, чтобы разобраться, почему она вообще возникла: это ожидаемое поведение при расчетах или же мы должны поправить сами расчеты?
Хотя бы простое:

for i in list(top_5_platforms['platform']):
    try:
        sns.lineplot(data=year_total_sale_for_platform(i, df)['total_sales'], label=i)
        plt.legend()
    except KeyError:
        # выводим текущие значения
        print('Ошибка при', i)
        print(year_total_sale_for_platform(i, df))

Тогда можно будет увидеть, при каком i возникает такая проблема, и куда в результате работы year_total_sale_for_platform девается ожидаемое значение ключа 'total_sales'.