Python 2D数组搜索行和列上的相邻单元格元素

基础概念

在Python中，2D数组通常指的是一个二维列表（list of lists），其中每个元素本身也是一个列表。这种数据结构可以用来表示矩阵或表格。

类型

整数数组：所有元素都是整数。
浮点数数组：所有元素都是浮点数。
混合类型数组：数组中的元素可以是不同类型。

应用场景

图像处理：2D数组可以用来表示图像的像素矩阵。
游戏开发：用于存储游戏地图、角色位置等信息。
数据分析：在统计和数据分析中，2D数组常用来表示数据表格。

问题：搜索行和列上的相邻单元格元素

假设我们有一个2D数组，并且我们想要找到某个单元格的相邻单元格（上、下、左、右）。

示例代码

def get_adjacent_cells(matrix, row, col):
    rows = len(matrix)
    cols = len(matrix[0])
    adjacent_cells = []

    # 上
    if row > 0:
        adjacent_cells.append((row - 1, col))
    # 下
    if row < rows - 1:
        adjacent_cells.append((row + 1, col))
    # 左
    if col > 0:
        adjacent_cells.append((row, col - 1))
    # 右
    if col < cols - 1:
        adjacent_cells.append((row, col + 1))

    return adjacent_cells

# 示例2D数组
matrix = [
    [1, 2, 3],
    [4, 5, 6],
    [7, 8, 9]
]

# 获取(1, 1)位置的相邻单元格
adjacent_cells = get_adjacent_cells(matrix, 1, 1)
print(adjacent_cells)  # 输出: [(0, 1), (2, 1), (1, 0), (1, 2)]