在指定的坐标和方向上从2D数组中获取n个元素

def get_elements_from_2d_array(array, start_coordinate, direction, n):
    result = []
    x, y = start_coordinate
    rows = len(array)
    cols = len(array[0])

    for _ in range(n):
        if x < 0 or x >= rows or y < 0 or y >= cols:
            break
        result.append(array[x][y])

        if direction == "up":
            x -= 1
        elif direction == "down":
            x += 1
        elif direction == "left":
            y -= 1
        elif direction == "right":
            y += 1

    return result