如何为NumPy矩阵添加‘重力’效果以实现四子棋落子？

你的代码思路方向是对的，但存在两处关键问题，导致逻辑无法正常工作：

问题分析 #

1. 列索引获取错误：row_gravity = game_board.T[player_row_move] 搞错了索引逻辑——你需要获取的是玩家选中的列（player_move）对应的所有行数据，而非用行索引player_row_move去取转置后的行。正确的列数据应该是game_board[:, player_move]（原棋盘直接取列）或者game_board.T[player_move]（转置后取行）。
2. 循环逻辑不完整：当前代码只执行一次下移判断，无法让棋子落到列的最底部空位。比如列内有多个连续空位时，代码只会下移一次，不会持续到最下方。

修正后的实现 #

方法1：循环下移到最底部 #

先获取玩家选中的列，从顶部初始行开始，循环检查下方是否为空，直到无法下移为止：

# 假设game_board是rows×cols维度的全零numpy数组
# player_move是用户输入的合法列索引（0到cols-1）

# 获取选中列的所有行数据
selected_col = game_board[:, player_move]
current_row = 0  # 初始行设为顶部

# 循环下移：只要下一行在棋盘范围内且为空，就继续下移
while current_row + 1 < game_board.shape[0] and selected_col[current_row + 1] == 0:
    current_row += 1

# 在最终位置放置玩家棋子（1代表玩家1，可根据需求替换为其他标识）
game_board[current_row, player_move] = 1
print(game_board)

方法2：直接定位最底部空位（更高效） #

利用numpy数组操作，直接找到列内最后一个空位的索引，避免循环：

import numpy as np

# 获取选中列数据
selected_col = game_board[:, player_move]
# 从下往上找第一个0的索引（即最底部的空位）
last_zero_idx = np.max(np.where(selected_col == 0)[0])

# 放置棋子
game_board[last_zero_idx, player_move] = 1
print(game_board)

额外注意事项 #

• 落子前要先校验玩家输入的列索引是否合法（在0到cols-1范围内），还要判断该列是否已满（无空位），避免数组越界或无效操作。
• 玩家状态切换（player_state）应在落子完成后统一执行，不要嵌套在落子判断逻辑里。

内容的提问来源于stack exchange，提问作者ElectroVader