Python. Задача 5-9

Дана таблица, которую нужно вывести:

a = [ [5, 6, 3, 8, 2], [3, 5, 9, 2, 8], [8, 3, 7, 5, 1] ] ]

Пользователь вводит два числа N, M < 5. Поменять в таблице столбцы N и M местами и вывести результат.

Решение
a = [ [5 ,6 ,3 ,8 , 2], [3, 5, 9, 2, 8], [8, 3, 7, 5, 1] ]
for i in range(3):
    for j in range(5):
        print('%2d' % a[i][j], end ='')
    print()
print()
N, M = map(int, input('Введите индексы N и M < 5: ').split())
for i in range(3):
    a[i][N], a[i][M] = a[i][M], a[i][N]
for i in range(3):
    for j in range(5):
        print('%2d' % a[i][j], end ='')
    print()