Получить строки из фрейма данных с помощью массива индекса из предложения numpy where

Question

У меня есть датафрейм с колонкой «Бюджет». Я хочу получить все строки из фрейма данных, значение которых соответствует столбцу бюджета 0. Я использую этот код:

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

Но я получаю это исключение:

IndexingError: Слишком много индексаторов

Answer 1

Вы хотите сделать boolean indexing:

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