вот код:
n, m = [int(i) for i in input().split()] matrix = [[0] * m for _ in range(n)] counter = 1 for i in range(n): for j in range(m): matrix[i][j] = counter print(str(matrix[j][i]).ljust(3), end=' ') counter += 1 print()
если 4 на 4 матрица выводит:
1 0 0 0
2 6 0 0
3 7 11 0
4 8 12 16
а нужно
1 5 9 13
2 6 10 14
3 7 11 15
4 8 12 16