파이썬 numpy.diag 함수 활용하기

Numpy Diag 함수 소개:

Numpy의 `numpy.diag` 함수는 주어진 배열의 대각선 요소를 추출하여 1차원 배열로 반환하거나, 주어진 1차원 배열을 대각선 요소로 갖는 대각행렬을 생성합니다. 이 함수는 주로 선형대수 연산에서 사용됩니다.

기본 사용법:

`numpy.diag` 함수의 기본 사용법은 다음과 같습니다.

import numpy as np

# 1차원 배열의 대각행렬 생성
array = np.array([1, 2, 3])
diag_matrix = np.diag(array)
print("diag_matrix:", diag_matrix)

위 예시 코드에서 `np.diag(array)`는 배열 `[1, 2, 3]`의 대각행렬을 생성합니다.

예시 코드:

• 2차원 배열의 대각선 요소 추출:

# 2차원 배열의 대각선 요소 추출
matrix = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
diagonal_elements = np.diag(matrix)
print("diagonal_elements:", diagonal_elements)

결론:

`numpy.diag` 함수는 배열의 대각 요소를 추출하거나 주어진 1차원 배열을 대각행렬로 변환하는 데 사용됩니다. 선형대수 및 통계 연산에서 다양한 용도로 활용할 수 있습니다. 위 예시 코드를 통해 `numpy.diag` 함수의 활용법을 익혀보세요.