帮我利用python写一个回形取数代码

下面是一个示例代码，使用了 Python 的 for 循环和 range 函数来实现回形取数：

n = int(input("请输入矩阵的大小(n*n)："))
matrix = [[0] * n for i in range(n)] # 创建一个 n*n 的矩阵

# 定义变量来记录当前填入数字的坐标
x, y = 0, 0
dx, dy = [0, 1, 0, -1], [1, 0, -1, 0] # 定义四个方向的移动
direction = 0 # 当前移动的方向

for i in range(1, n * n + 1):
    matrix[x][y] = i # 在当前坐标填入数字
    if x + dx[direction] < 0 or x + dx[direction] >= n or y + dy[direction] < 0 or y + dy[direction] >= n or matrix[x + dx[direction]][y + dy[direction]] != 0:
        # 如果移动后越界或者下一个位置已经有数字了，那么更换移动方向
        direction = (direction + 1) % 4
    x += dx[direction]
    y += dy[direction]

# 打印结果
for row in matrix:
    print(row)

这段代码演示了一个回形取数问题的简单实现。首先，程序会请求用户输入一个整数 n，表示矩阵的大小。接着，程序会创建一个 n*n 的矩阵，用于存储回形取数的数字。然后，程序会使用 for 循环来填入数字，并使用 dx 和 dy 数组来控制移动方向。最后，程序会打印出结果。