s = [[10, 20, 30], [30, 40, 50], [50, 60, 70], [70, 80, 90]] print("2차원 배열 :", s) print("열에 따른 산술 평균 :", np.mean(s, axis=0)) print("행에 따른 산술 평균 :", np.mean(s, axis=1))
실행 결과
2차원 배열 : [[10, 20, 30], [30, 40, 50], [50, 60, 70], [70, 80, 90]] 열에 따른 산술 평균 : [40. 50. 60.] 행에 따른 산술 평균 : [20. 40. 60. 80.]
np.where()는 괄호 안 조건에 맞는 값을 찾아서 그 원소의 인덱스를 배열로 반환한다. 다음 코드처럼 a < 3라고 사용하면 3보다 작은 숫자가 첫 번째 원소와 두 번째 원소임을 알려 준다.
np.where(a < 3)
실행 결과
(array([0, 1]),)