Получение KeyError при группировании моего набора данных в 2 образца

Question

Я прохожу онлайн-курс. «bikesharing_data» - это имя объекта pandas, а «рабочий день» - это имя столбца в этом фрейме данных. Репетитор хочет разделить набор данных на две выборки и разделить «рабочий день» на ([0, 1]) группы. Вот что она написала:

sample_01 = bikesharing_data[(bikesharing_data['workingday'] == 1)]

sample_02 = bikesharing_data[(bikesharing_data['workingday'] == 0)]

Мой вопрос, если я разделю образец следующим образом:

sample_01 = bikesharing_data['workingday' == 1]

, это дает мне KeyError. Я понимаю, что такое KeyError, но почему он выдает ошибку ключа?

Answer 1

'workingday' == 1

Это дает логическое значение. Так что для вашего примера это будет эквивалентно вызову sample_01 = bikesharing_data[False]'

, поскольку строка не равна 1. Если False не является ключом в вашем наборе, вы получите KeyError. Возможно, вы хотите использовать индекс в качестве ключа, а не логическое значение.