将我的数据集分组为 2 个样本时出现 KeyError

Question

我正在上在线课程。 'bikesharing_data' 是 pandas 对象的名称，'workingday' 是该数据框中列的名称。导师想把数据集分成两个样本，把'workingday'分成([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。您可能希望使用索引作为键，而不是布尔值。