Problem

8 /12


Matriz diagonal

Theory Click to read/hide

Matrizes diagonais

A função  diag(V, k=0) permite extrair uma diagonal de uma matriz, bem como construir matrizes diagonais a partir de matrizes unidimensionais.
V - Um objeto tipo array, arrays bidimensionais ou unidimensionais, matrizes, listas ou tuplas, ou qualquer função ou objeto com um método que retorna uma lista ou tupla.
k - índice da diagonal (opcional).
O padrão é k = 0 que corresponde à diagonal principal. Um valor k positivo move a diagonal para cima, um valor negativo a move para baixo.

A função retorna array NumPy (ndarray) - a diagonal da matriz especificada ou uma matriz diagonal da matriz unidimensional especificada.

Problem

A entrada é o número n. Gera uma matriz de tamanho nxn, in in os números diagonais são 0 a n-1 e o resto dos números são 0 .
 

 

Exemplos
# Entrada Saída
1 5 [[0 0 0 0 0]
 [0 1 0 0 0]
 [0 0 2 0 0]
 [0 0 0 3 0]
 [0 0 0 0 4]]

time 1000 ms
memory 256 Mb
Rules for program design and list of errors in automatic problem checking

Teacher commentary

Your Program

Attach a file with the source code of the program:

or select a programming language and type your program

Auto

       

Program check result To check the solution of the problem, you need to register or log in!