Как создать единичную матрицу на Python, если на главной диагонали НЕ единицы?

Question

[Timebird]

Здравствуйте!
На python вот потребовалось создать единичную матрицу, но так, чтобы по главной диагонали были не единицы, а значения из массива (массив известен, в нём 80 элементов, допустим).

Для обычной единичной матрицы необходимо применить numpy.eye(80). А что делать в моём случае?
Спасибо!

Comments

[Владимир Олохтонов]

Такая матрица называется диагональной :) Ответ Vlad_Fedorenko выглядит наиболее разумным. Вот ссылка на документацию по этой функции: https://docs.scipy.org/doc/numpy/reference/generat...

Answer 1

[Vlad_Fedorenko]

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

Answer 2

[NaName]

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

Comments

[AlexKuznec]

Диагональные матрицы хранятся в виде одномерных массивов, так как кучу нулей хранить и использовать в вычислениях крайне неэффективно.
Фактически, input_massive и должен использоваться)

[NaName]

AlexKuznec: понял, фигню написал. Спасибо.