Numpy NP.diag

A função Numpy Diag () permite extrair ou criar uma nova matriz diagonal. É uma função conveniente, mas direta.

Siga junto com este tutorial para explorar ainda mais esta função.

Função Sintaxe

A sintaxe da função é como mostrado abaixo:

Numpy.diag (v, k = 0)

A função leva apenas dois parâmetros, como discutido abaixo:

Parâmetros de função

1. V - A matriz de entrada ou objeto ARRAY_LIE.
2. K - define a diagonal a ser extraída.

Nota: Se o valor de k for maior que 0, significa a diagonal acima da diagonal principal. Se for negativo, significa a diagonal abaixo da diagonal principal.

Valor de retorno da função

A função retorna a matriz diagonal extraída ou uma matriz diagonal recém -construída.

Exemplo 1

Considere o código de exemplo mostrado abaixo:

importar numpy como np
arr = np.ARANGE (6).remodelar (2,3)
Imprimir (arr)
Imprimir (NP.diag (arr, k = 0))

Usamos a função diag () para extrair a diagonal principal da matriz fornecida no código acima.

A saída resultante é mostrada:

[[0 1 2]
[3 4 5]]
--> Extraído: [0 4]

Exemplo 2

Para extrair a diagonal acima da principal, defina o valor de k como mostrado:

arr = np.ARANGE (6).remodelar (2,3)
print (f "original: arr")
Print (F "Extrato: NP.diag (arr, k = 1) ")

Isso retorna:

Original: [[0 1 2]
[3 4 5]]
Extrato: [1 5]

Exemplo 3

Se o valor de k for negativo, deve retornar:

arr = np.ARANGE (6).remodelar (2,3)
print (f "original: arr")
Print (F "Extrato: NP.diag (arr, k = -1) ")

Saída:

Original: [[0 1 2]
[3 4 5]]
Extrato: [3]

Conclusão

Agora você está familiarizado com a função diag em Numpy e como usá -la para extrair ou construir uma nova matriz diagonal.

Obrigado pela leitura!!