본문 바로가기

Python/기초 문법

[모듈 2-7] NUMPY 기초 사용법

import numpy as np


### 1. Numpy 배열 만들기 - 리스트 넘파이 배열로 변환

list1 = [1, 2, 3, 4]
a = np.array(list1)
print(a.shape) # (4, )

b = np.array([[1, 2, 3], [4, 5, 6]])
print(b.shape) # (2, 3)
print(b[0, 0]) # 1


### 2. 넘파이 배열 선언 함수 사용하기
a = np.zeros((2, 2))
print(a)
# 출력:
# [[ 0. 0.]
# [ 0. 0.]]

a = np.ones((2, 3))
print(a)
# 출력:
# [[ 1. 1. 1.]
# [ 1. 1. 1.]]

a = np.full((2, 3), 5)
print(a)
# 출력:
# [[5 5 5]
# [5 5 5]]

a = np.eye(3)
print(a)
# 출력:
# [[ 1. 0. 0.]
# [ 0. 1. 0.]
# [ 0. 0. 1.]]

a = np.array(range(20)).reshape((4, 5))
print(a)
# 출력:
# [[ 0 1 2 3 4]
# [ 5 6 7 8 9]
# [10 11 12 13 14]
# [15 16 17 18 19]]

### 3. numpy 배열 슬라이싱

lst = [
[1, 2, 3],
[4, 5, 6],
[7, 8, 9]
]
arr = np.array(lst)

# 슬라이스
a = arr[0:2, 0:2]
print(a)
# 출력:
# [[1 2]
# [4 5]]

a = arr[1:, 1:]
print(a)
# 출력:
# [[5 6]
# [8 9]]

### 4. numpy 배열 인덱싱


lst = [
[1, 2, 3, 4],
[5, 6, 7, 8],
[9, 10, 11, 12]
]
a = np.array(lst)

# 정수 인덱싱
s = a[[0, 2], [1, 3]]

print(s)
# 출력
# [2 12]


### 5. numpy 배열 연산

a = np.array([1, 2, 3])
b = np.array([4, 5, 6])

# 각 요소 더하기
c = a + b
# c = np.add(a, b)
print(c) # [5 7 9]

# 각 요소 빼기
c = a - b
# c = np.subtract(a, b)
print(c) # [-3 -3 -3]

# 각 요소 곱하기
# c = a * b
c = np.multiply(a, b)
print(c) # [4 10 18]

# 각 요소 나누기
# c = a / b
c = np.divide(a, b)
print(c) # [0.25 0.4 0.5]


# 행렬의 합 구하기
a = np.array([[1, 2], [3, 4]])

s = np.sum(a)
print(s) # 10

# axis=0 이면, 컬럼끼리 더함
# axis=1 이면, 행끼리 더함
s = np.sum(a, axis=0)
print(s) # [4 6]

s = np.sum(a, axis=1)
print(s) # [3 7]

s = np.prod(a)
print(s) # 24