python - 使用 numpy where 子句中的索引数组从数据框中获取行

Question

我有一个数据框，其中有一列“预算”。我想从数据框中获取与预算列对应的值为0的所有行。我使用以下代码：

dataFrameTrain.iloc[np.where(dataFrameTrain['budget']==0),:]

但是，我得到了这个例外：

IndexingError：索引器过多

标签： pythonpandasnumpy

Answer 1

你想做一个boolean indexing：

dataFrameTrain[dataFrameTrain['budget']==0]