Создание матрицы с заполнением построчно

Необходимо создать матрицу размером N×M и заполнить ее числами от 1 до N×M по возрастанию слева направо, сверху вниз (построчно).

N = int(input('Введите количество строк в матрице: '))
M = int(input('Введите количество столбцов: '))
matrix = [[0] * M for i in range(N)]
count = 1
for i in range(N):
    for j in range(M):
        matrix[i][j] = count
        count += 1
for row in matrix:
    print(*row)