python foreach array with index