统计python中二维数组中相邻相等元素的个数

def count_adjacent_equal_elements(matrix):
    count = 0
    rows = len(matrix)
    columns = len(matrix[0])

    # 检查水平方向上的相邻元素
    for i in range(rows):
        for j in range(columns - 1):
            if matrix[i][j] == matrix[i][j+1]:
                count += 1

    # 检查垂直方向上的相邻元素
    for i in range(rows - 1):
        for j in range(columns):
            if matrix[i][j] == matrix[i+1][j]:
                count += 1

    return count