number of columns and rows python