递增2D列表中的一个正方形，同时递减所有其他不涉及的正方形- Python

def update_square(matrix, row, col, size):
    # 递增指定正方形
    for i in range(row, row+size):
        for j in range(col, col+size):
            matrix[i][j] += 1
    
    # 递减其他正方形
    for i in range(len(matrix)):
        for j in range(len(matrix[i])):
            if i < row or i >= row+size or j < col or j >= col+size:
                matrix[i][j] -= 1
    
    return matrix