牛客题解 | 将向量转换为对角矩阵

题目

题目链接

将向量转换为对角矩阵（Diagonal Matrix）是一种常用的矩阵，其计算公式为：

\[D = \begin{bmatrix} a_1 & 0 & 0 \\ 0 & a_2 & 0 \\ 0 & 0 & a_3 \end{bmatrix} \]

其中，\(a\)是向量。
通俗的说，就是将向量中的每个元素作为对角矩阵的对角线上的元素。

标准代码如下

def make_diagonal(x):
    identity_matrix = np.identity(np.size(x))
    return (identity_matrix*x)