Numpy arange, reshape란

안녕하세요 jay입니다.

오늘은 numpy arange와 reshape 함수에 대해 알아보도록 하겠습니다.

이 두 함수는 같이 쓰기 유용한 함수입니다.

arange() : 주어진 범위만큼, 1차원 array를 만드는 함수입니다. 끝 인덱스는 포함하지 않는다.

reshape() : 기존의 array 형태를 다시 바꿔주는 함수입니다.

import numpy as np

# arange : parameter의 범위만큼의 1차원 array를 만드는 함수, 끝값은 포함을 하지 않는다
arr = np.arange(1,10)
print(arr) # [1 2 3 4 5 6 7 8 9]

# reshape : array의 형태를 바꿔주는 함수 (row,column) 형태로 바뀜

new_arr = arr.reshape(3,3)
print(new_arr)
"""
[[1 2 3]
 [4 5 6]
 [7 8 9]]
"""

# reshape를 한 뒤, 새로운 변수에 저장해야합니다.

arr.reshape(9,1)
print(arr) # [1 2 3 4 5 6 7 8 9]

# 한줄로도 표현이 가능합니다. 이 형태를 더 많이 사용합니다.
arr2 = np.arange(1,17).reshape(4,4)
print(arr2)
"""
[[ 1  2  3  4]
 [ 5  6  7  8]
 [ 9 10 11 12]
 [13 14 15 16]]
"""