Как удалить строки из фрейма данных?

Question

Я пытаюсь удалить выбросы из набора данных. Для того, чтобы сделать это, я использую:

df = df[df.attr < df.attr.mean() + df.attr.std()*3] 

Это, кажется, работает, как ожидалось, но, когда я делаю что-то вроде:

for i in xrange(df.shape[0]): 
    print df.attr[i] 

Тогда я получаю KeyError. Кажется, что Pandas фактически не возвращает новый DataFrame с опущенными рядами. Как я могу удалить эти строки и получить полностью функциональный DataFrame?

Answer 1

Во-первых, найти индексы, которые удовлетворяют критериям (которые в ваш случай - df.attr < df.attr.mean() + df.attr.std() * 3).

x = df.loc[:,attr] < df.attr.mean() + df.attr.std()*3 

Дальше, используйте DataFrame.drop.

df.drop(x[x].index) 

См ответы, такие как How to drop a list of rows from Pandas dataframe? для получения дополнительной информации

Answer 2

Я думаю, что нужно DataFrame.ix:

for i in xrange(df.shape[0]): 
    print df.ix[i, 'attr'] 

Или Series.iloc:

for i in xrange(df.shape[0]): 
    print df.attr.iloc[i] 

решение проще с Series.iteritems:

for i, val in df.attr.iteritems(): 
    print (val)