生成指定元素的对角矩阵

生成指定元素的对角矩阵可以使用以下代码实现：

import numpy as np

def diagonal_matrix(elements):
    """
    生成指定元素的对角矩阵
    
    :param elements: 一维数组，指定对角线上的元素
    :return: 生成的对角矩阵
    """
    n = len(elements)
    matrix = np.zeros((n, n))
    for i in range(n):
        matrix[i][i] = elements[i]
    return matrix

其中，`elements` 是一维数组，表示对角线上的元素。函数会返回一个 numpy 数组，表示生成的对角矩阵。