python get list index in for loop