머신러닝/넘파이

[Numpy] 리스트 ndarray 변환 함수 - tolist(), np.array()

shoney9254 2021. 8. 23. 21:13
반응형

급한 상황에서, 별거아닌 함수도 기억 못하는 경우가 있다.(나만 있나?) 

그런 사람들을 위해서 엄청 간단한 함수에 대해서 정리해봤다.

 

다시 말하지만, 이번 포스팅은 엄청 별거 없다. 

 

먼저, zeros로 0으로 (3,3) shape ndarray 를 생성한다. 

소스 코드

array2d = np.zeros(shape=(3, 3), dtype='int32')

print('array2d:\n', array2d)

결과

array2d:
 [[0 0 0]
 [0 0 0]
 [0 0 0]]

 

1. tolist() 함수

ndarray 에서 list로 변경 

소스 코드

# tolist() 는 ndarray를 list로 변환 해준다.

array1 = array2d.tolist()

print('array1:\n', array1)

print('type:', type(array1))

결과

array1:
 [[0, 0, 0], [0, 0, 0], [0, 0, 0]]
type: <class 'list'>

 

2. nd.array() 함수

list에서 ndarray 로 변경해준다. 

소스 코드

ndarray1 = np.array(array1)

print('ndarray1:\n', ndarray1)

print('type:', type(ndarray1))

결과

ndarray1:
 [[0 0 0]
 [0 0 0]
 [0 0 0]]
type: <class 'numpy.ndarray'>

 

반응형