python code for rows and columns