one line for loop python