np.diag() 是 NumPy 库中的一个函数,用于创建一个对角矩阵或从一个矩阵中提取对角线元素.具体来说,np.diag() 可以接受以下两种参数: 一个一维数组,将其转换为一个对角矩阵.数组中的元素将成为对角线上的元素.例如,np.diag([1,2,3]) 将创建一个三阶的对角矩阵,对角线上的元素分别为 1,2,3. 一个二维数组,将其对角线上的元素提取出来,返回一个一维数组.例如,np.diag([[1,2,3],[4,5,6],[7,8,9]]) 将返回一个一维数组,包含矩阵 [[