Numpy NP.Diag

Funkcja Numpy Diag () pozwala wydobyć lub utworzyć nową tablicę przekątną. Jest to wygodna, ale prosta funkcja.

Postępuj zgodnie z tym samouczkiem, aby dalej zbadać tę funkcję.

Składnia funkcji

Składnia funkcji jest jak pokazana poniżej:

Numpy.Diag (v, k = 0)

Funkcja przyjmuje tylko dwa parametry, jak omówiono poniżej:

Parametry funkcji

1. v - tablica wejściowa lub obiekt podobny do tablicy.
2. k - definiuje przekątkę do wyodrębnienia.

Uwaga: Jeśli wartość K jest większa niż 0, oznacza to przekątna powyżej głównej przekątnej. Jeśli jest ujemny, oznacza to przekątna poniżej głównej przekątnej.

Funkcja zwraca wartość

Funkcja zwraca wyodrębnioną tablicę przekątną lub nowo skonstruowaną tablicę przekątną.

Przykład 1

Rozważ przykładowy kod pokazany poniżej:

importować Numpy jako NP
ARR = NP.Arange (6).Reshape (2,3)
Drukuj (ARR)
Drukuj (NP.Diag (ARR, k = 0))

Używamy funkcji diag () do wyodrębnienia głównego przekątnego z dostarczonej tablicy w powyższym kodzie.

Pokazano wynikowe wyjście:

[[0 1 2]
[3 4 5]]
--> Wyodrębniono: [0 4]

Przykład 2

Aby wyodrębnić przekątną powyżej głównego, ustaw wartość K jako jedną, jak pokazano:

ARR = NP.Arange (6).Reshape (2,3)
print (f "oryginał: arr")
druk (f "Extract: np.diag (arr, k = 1) ")

To powraca:

Oryginał: [[0 1 2]
[3 4 5]]
Ekstrakt: [1 5]

Przykład 3

Jeśli wartość K jest ujemna, powinna zwrócić:

ARR = NP.Arange (6).Reshape (2,3)
print (f "oryginał: arr")
druk (f "Extract: np.diag (arr, k = -1) ")

Wyjście:

Oryginał: [[0 1 2]
[3 4 5]]
Ekstrakt: [3]

Wniosek

Teraz znasz funkcję diag w Numpy i jak ją używać do wydobywania lub budowy nowej tablicy przekątnej.

Dziękuje za przeczytanie!!