添加函数以编辑二维数组中对角线上的数字

def edit_diagonal_numbers(matrix, new_value):
    """
    编辑二维数组中对角线上的数字为新的值

    参数：
    matrix: 二维数组
    new_value: 新的值

    返回值：
    编辑后的二维数组
    """

    # 获取二维数组的行数和列数
    rows = len(matrix)
    cols = len(matrix[0])

    # 编辑主对角线上的数字
    for i in range(min(rows, cols)):
        matrix[i][i] = new_value

    # 编辑副对角线上的数字
    for i in range(min(rows, cols)):
        matrix[i][cols-i-1] = new_value

    return matrix