본문 바로가기
대학원 공부/programming language

Numpy : np.reshape, flatten, ravel, newaxis (차원 변경)

by 월곡동로봇팔 2020. 2. 21.

reshape

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

A.reshape(3,2)
# A.reshape(-1,1) 위와 같은 것, -1은 열의 갯수에 맞춰서 행 갯수를 자동으로 정하는 것.
"""
[[1,2],
 [3,4],
 [5,6]]
"""

flatten, ravel

A.flatten()
# A.ravel()
A
A.shape
"""
[1,2,3,4,5,6]
"""

newaxis

A[:, np.newaxis]

"""
array([[1],
	   [2],
       [3],
       [4],
       [5],
       [6]])
"""

meshgrid

points = np.arange(-5,5,0.01)

xs,ys = np.meshgrid(points, points)

xs
"""
[[-5, -4.99 ------- 4.99, 4.99],
 [-5, -4.99 ------- 4.99, 4.99],
 [-5, -4.99 ------- 4.99, 4.99],
 [-5, -4.99 ------- 4.99, 4.99],
 ----
 [-5, -4.99 ------- 4.99, 4.99]]
"""

ys
"""
[[-5, -5 ------- -5, -5],
 [-4.99, -4.99 ---- -4.99, -4.99],
 ----
 [4.99, 4.99 ----- 4.99, 4.99],
 [5,5,5,5,------5,5]]
"""

meshgrid의 xs는 1차원행렬을 하나의 행으로 정방행렬을 만든다.

meshgrid의 ys는 1차원행렬을 하나의 열으로 정방행렬을 만든다.

댓글