python check if element in list and get index