python for loop return index and value