Python Keep Rows Based On Condition Maintenance