Матрицы — обмен максимального и минимального в строке

Матрица вводится пользователем. Далее необходимо в каждой строке поменять местами элементы с минимальным и максимальным значением. Вывести новую матрицу.

n = int(input("Введите размер матрицы: "))
matrix = [[0] * n for i in range(n)]
for i in range(n):
    matrix[i] = [int(i) for i in input("Введите строку матрицы: ").split()]
for row in matrix:
    min_elem = min(row)
    max_elem = max(row)
    min_index = row.index(min_elem)
    max_index = row.index(max_elem)
    row[min_index], row[max_index] = row[max_index], row[min_index]
print("Измененная матрица: ")
for row in matrix:
    print(*row)