Создать программу с присваиванием каждому элементу матрицы значение некоторого выражения (небольшую)

питон

n = int(input('введите размер матрицы '))

arr = [[0] * n for i in range(n)]

for i in range(n):

for j in range(n):

arr[i][j] = n * i + j + 1

print('n'. join([''. join([str(f'{i:3}') for i in row]) for row in arr]))