import operator
n, k = int(input()), int(input()) # ввод n и k
a = [[int(input()) for i in range(n)] for j in range(n)] # ввод матрицы a
a.sort(key=operator.itemgetter(k - 1)) # и это вся сортировка по столбцу k
for r in a: print(*r) # вывод матрицы
Подробности:
https://tproger.ru/translations/python-sorting/