if element in list python get index