首页 > 试题广场 >

将向量转换为对角矩阵

[编程题]将向量转换为对角矩阵
  • 热度指数:823 时间限制:C/C++ 1秒,其他语言2秒 空间限制:C/C++ 256M,其他语言512M
  • 算法知识视频讲解
实现一个函数,将一维向量转换为对角矩阵。对角矩阵是一个方阵,其主对角线上的元素来自输入向量,而其他位置的元素都为0。

输入描述:
第一行输入一个一维向量,长度为 n


输出描述:
返回一个二维numpy数组(方阵):
- 维度为 n \times n,其中 n 是输入向量的长度
- 对角线上的元素来自输入向量
- 其他位置的元素都为0
元素类型不需要手动修改,让它为默认的即可。
示例1

输入

[1, 2, 3]

输出

[[1. 0. 0.]
 [0. 2. 0.]
 [0. 0. 3.]]

备注:
1.对应的输入、输出已给出,您只用实现核心功能函数即可。
2.支持numpy、scipy、pandas、scikit-learn库。
校验格式真的很无语
发表于 2025-11-05 11:05:06 回复(0)