행렬에서 max값 가져오기

import numpy as np

a = np.array([[0, 1, 2, 4], [5, 6, 7, 8], [9, 0, 1, 2]])
print(a)

b = np.max(a, axis=0)
print(b)

c = np.max(a, axis=1)
print(c)

d = np.max(a, axis=1, keepdims=True)
print(d)

#print(a)
[[0 1 2 4]
 [5 6 7 8]
 [9 0 1 2]]
 
 print(b)
 [9 6 7 8]
 
 print(c)
 [4 8 9]
 
 print(d)
 [[4]
 [8]
 [9]]

a: 사용할 행렬
b: axis=0일 때 1열, 2열,...,n열 확인
c: axis=1일 때 1행, 2행,...,n행 확인
d: axis=1, keepdims=True일 때 차원 형태 유지