Как создать диагональную матрицу с более коротким списком диагоналей в 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)