python for loop range len if == comparison function