[6기] 프로그래머스 인공지능 데브코스 34일차 TIL

1004

7주차-Day3) Machine Learning 기초

선형대수 실습

실습 문제에 사용된 코드들 공유

import numpy as np

I = np.eye(5, dtype = np.float32)

print(I)

---

import numpy as np

A = [[1, 2, 3, 4, 5], [6, 7, 8, 9, 10], [11, 12, 13, 14, 15], [16, 17, 18, 19, 20], [21, 22, 23, 24, 25]]

A = np.array(A)
print(A)

# A의 대각 요소들을 추출한 결과를 변수 result1에 담아주세요.
result1 = 
np.diag(A)

print(result1)

# A의 대각 원소들을 모두 더한 값을 result2에 담아주세요.
result2 = 
np.trace(A)

print(result2)