Как разделить фрейм данных по значению столбца

Question

У меня есть фрейм данных с именем raw, и он выглядит следующим образом:

Я пытаюсь разделить основной фрейм данных на два отдельных фрейма данных (например, df1 и df2) в зависимости от того, является ли BDA нулевым или нет. Как я могу сделать это при сохранении других строк?

Answer 1

Вы можете сделать с groupby

df1,df2=[x for _,x in df.groupby('BDA')]

Answer 2

Если у вас есть фрейм данных df, вы можете использовать простую нарезку:

df1 = df[df['BDA'] = 'null']
df2 = df[df['BDA'] = 'not null']

Это не повлияет на ваш исходный фрейм данных.