How to create an identity matrix in Python if there are NOT ones on the main diagonal?

Question

Python

Hello!
In python, it was required to create an identity matrix, but in such a way that along the main diagonal there were not units, but values ​​from the array (the array is known, it has 80 elements, let's say).
For an ordinary identity matrix, numpy.eye(80) must be applied. What should I do in my case?
Thank you!

Answer 1

np.diag([1,2,3])

Answer 2

#инитиализация нулевой матрицы нужной размерности
matrix = [...][...]
#инитиализация массива значений диагонали
input_massive = [...]
#заполняем диагональ
for i in range(0, len(input_massive)):
    matrix[i][i] = input_massive[i]