結果 : python split array into two based on condition