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

Question

У меня есть этот фрейм данных:

В строке 84 он меняется с отрицательного значения в столбце «позиция» на положительное.

Мне нужно в этот момент разбить фрейм данных на два (до и после позиции = 0)

Я пытался использовать

idx = run_1[run_1['Position_(m)']>0].index dfs = np.split(df, idx)

но он разделяется на каждый экземпляр, где> 0, и если я использую idx [0], он говорит:

ValueError: разбиение массива не приводит к равному делению

Примечание: мне нужно только одно разбиение, «до и после», я просмотрел другие подобные вопросы, но большинство из них касается нескольких разделений, и я не могу повторно использовать код, предложенный для одного разделения.

Answer 1

Вы можете использовать логическую маску, чтобы сделать это:

before = run_9[run_9['Position_(m)']<0]
after = run_9[run_9['Position_(m)']>0]