用python实现方格取数

在Python中，你可以通过列表推导式或者嵌套循环来实现一个简单的方格取数游戏。这个游戏通常是在一个二维数组（比如一个n x n的矩阵）中，玩家需要按特定规则取数，例如每次只能向右、向下、向右下或左下走一步，并且取出的数字会被添加到总和中。

下面是一个基本的例子，创建一个5x5的矩阵并模拟玩家的移动：

# 初始化一个5x5的矩阵
grid = [[0] * 5 for _ in range(5)]
grid[0][0] = 1

# 设置初始位置和游戏结果变量
position = [0, 0]
total_sum = 0

# 方格取数函数
def take_number(direction):
    global position, grid, total_sum
    dx, dy = { 'right': (0, 1), 'down': (1, 0), 'right_down': (1, 1), 'left_down': (-1, 1) }[direction]
    new_position = [pos + step for pos, step in zip(position, (dx, dy))]
    
    if 0 <= new_position[0] < len(grid) and 0 <= new_position[1] < len(grid[0]):
        total_sum += grid[new_position[0]][new_position[1]]
        grid[new_position[0]][new_position[1]] = 0
        position = new_position
    else:
        print("越界了，无法移动")

# 示例玩家操作
take_number('right')  # 向右
take_number('down')   # 向下
take_number('right_down')  # 右下