Условие

Дан двумерный массив и два числа: i и j. Поменяйте в массиве столбцы с номерами i и j и выведите результат.

Программа получает на вход размеры массива n и m, затем элементы массива, затем числа i и j.

Решение оформите в виде функции swap_columns(a, i, j).

Решение

1. def swap_columns(a, i, j):
2.     for k in range(len(a)):
3.         a[k][i], a[k][j] = a[k][j], a[k][i]
4. n, m = [int(i) for i in input().split()]
5. a = [[int(j) for j in input().split()] for i in range(n)]
6. i, j = [int(i) for i in input().split()]
7. swap_columns(a, i, j)
8. print('\n'.join([' '.join([str(i) for i in row]) for row in a]))