Как удалить строки из DataFrame

Question

У меня есть DataFrame с n строками и ndarray с n значениями (-1 для выбросов и 1 для включения). Есть ли питонный способ удаления строк DataFrame, которые соответствуют индексам элементов nparray, помеченных как -1?

Answer 1

Вы можете просто сделать: new_df = old_df[arr == 1].

Пример

df = pd.DataFrame(np.random.randn(5,5))

arr = np.random.choice([1,-1], 5)
>>> df
          0         1         2         3         4
0 -0.238418  0.291475  0.139162 -0.030003 -0.515817
1 -0.162404 -1.272317  0.342051 -0.787938  0.464699
2 -0.965481  0.727143 -0.887149 -0.430592 -2.074865
3  0.699129 -0.242738  1.754805 -0.120637 -1.536973
4  0.228538  0.799445 -0.217787  0.398572 -1.255639
>>> arr
array([ 1, -1, -1,  1, -1])

>>> df[arr == 1]
          0         1         2         3         4
0 -0.238418  0.291475  0.139162 -0.030003 -0.515817
3  0.699129 -0.242738  1.754805 -0.120637 -1.536973