Python KeyErrors

Question

Это мой код:

def get_data(file_name):
    data = pd.read_csv(file_name, error_bad_lines=False, encoding='latin-1')
    X_parameter = []
    Y_parameter = []
    for single_square_feet, single_price_value in zip(data['Square_Feet'], data['Price']):
        X_parameter.append([float(single_square_feet)])
        Y_parameter.append(float(single_price_value))
    return X_parameter, Y_parameter

X, Y = get_data('F:\Demo\input_data.csv')

print(X)

print(Y)

Это ошибка:

File "pandas\_libs\index.pyx", line 140, in pandas._libs.index.IndexEngine.get_loc
File "pandas\_libs\index.pyx", line 162, in pandas._libs.index.IndexEngine.get_loc
File "pandas\_libs\hashtable_class_helper.pxi", line 1492, in pandas._libs.hashtable.PyObjectHashTable.get_item
File "pandas\_libs\hashtable_class_helper.pxi", line 1500, in pandas._libs.hashtable.PyObjectHashTable.get_item
KeyError: 'Square_Feet'

И мой .csv это:

Answer 1

У вас есть ошибка ключа в data['Square_feet'], что означает, что ваш ввод в качестве ключа не соответствует ни одному имени столбца.Я полагаю, что у вас есть некоторые пробелы.Поэтому, чтобы решить эту проблему, добавьте sep='\s*,\s*' к вашему pd.read.csv

data = pd.read_csv(file_name, sep='\s*,\s*', error_bad_lines=False, encoding='latin-1')