Задача "Поменять столбцы"

Условие

Дан двумерный массив и два числа: 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]))