下面是一个示例代码，演示了如何将一个NxN的数组的对角线元素作为一个数组返回。

def extract_diagonal_elements(matrix):
    # 获取数组的行数和列数
    n = len(matrix)
    m = len(matrix[0])
    
    # 确保输入的数组是一个NxN的矩阵
    if n != m:
        raise ValueError("输入的数组不是一个NxN的矩阵")
    
    # 创建一个空数组来存储对角线元素
    diagonal_elements = []
    
    # 使用两个循环遍历矩阵的元素
    for i in range(n):
        for j in range(m):
            # 如果元素的行索引和列索引相等，表示该元素在对角线上
            if i == j:
                diagonal_elements.append(matrix[i][j])
    
    # 返回对角线元素数组
    return diagonal_elements

# 测试代码
matrix = [[1, 2, 3],
          [4, 5, 6],
          [7, 8, 9]]

diagonal_elements = extract_diagonal_elements(matrix)
print(diagonal_elements)

输出：

[1, 5, 9]

这个示例代码中，我们首先获取了输入矩阵的行数和列数，然后使用两个循环遍历矩阵的元素。如果元素的行索引和列索引相等，表示该元素在对角线上，我们将其添加到对角线元素数组中。最后，我们返回对角线元素数组。