python - DataFrame.drop 不会在 Pandas 中删除预期的行

Question

我有一个 Pandas DataFrame，其中包含我要根据“人口”列中的值删除的行：

data['population'].value_counts()

general population                          21
developmental delay                         20
sibling                                      2
general population + developmental delay     1
dtype: int64

在这里，我想删除sibling作为值的两行。所以，我相信以下应该可以解决问题：

data = data.drop(data.population=='sibling', axis=0)

正如您在结果值计数中看到的那样，它确实删除了 2 行，但它们不是具有指定值的行。

data.population.value_counts()

developmental delay                         20
general population                          19
sibling                                      2
general population + developmental delay     1
dtype: int64

知道这里发生了什么吗？

Answer 1

dataFrame.drop接受索引（标签列表）作为参数，而不是掩码。
要使用drop你应该这样做：

data = data.drop(data.index[data.population == 'sibling'])

但是做起来要简单得多

data = data[data.population != 'sibling']