Python Loop Through Array Starting At Square