Эквивалент диаг функция R в Numpy питона

Question

Когда я с R diag(0.03,2) я получаю:

 [,1] [,2] 
[1,] 0.03 0.00 
[2,] 0.00 0.03 

Как я могу иметь один и тот же результат в Python? Я попробовал numpy.diag(0.03,2), но получаю ошибку: ValueError: Input must be 1- or 2-d. То же самое с numpy.diagonal(0.03,2)

Спасибо за любую помощь

Answer 1

Do numpy.diag([.03]*2).

numpy.diag() принимает список диагональных компонентов, и вы можете перемножать списки в python для повторения элементов.