Python Keep Rows Based On Condition Definition