python foreach array index