python numpy创造对角阵，且对角线上的元素自己定义

您可以使用numpy库中的diag函数来创建对角阵，并将对角线上的元素定义为自己想要的值。以下是一个示例代码：

import numpy as np

# 定义对角线上的元素
values = [1, 2, 3, 4]

# 创建对角阵
diagonal_matrix = np.diag(values)

print(diagonal_matrix)

输出结果将会是一个4x4的对角阵，对角线上的元素分别为1、2、3、4。您可以根据自己的需求修改values列表中的元素来定义不同的对角线上的值。