如何在Python中创建一个对角线列表较短的对角线矩阵？

Question

假设我有一个对角线条目列表。d有没有一种简单的方法来创建一个对角线矩阵，如 A? 我可以用 np.diag(d)?

import numpy as np

d = np.array([1,2])

A = np.array([[1,0,0,0], [0,2,0,0], [0,0,0,0], [0,0,0,0]])

Answer 1

这很简单。

A = np.zeros((4,4))
A[:2, :2] = np.diag(d)