python alternative to for loop