Remove elements from list based on condition python