본문 바로가기

BLOG/파이썬 기초

[파이썬] 리스트를 numpy 배열(array)로 바꾸려면

 

파이썬에서 numpy는 list 형식의 데이터를 다루지 못하고, array만 다룰 수 있는데요,

만약 리스트로 배열간 곱셈, 합 등 numpy 연산을 하고 싶다면 리스트를 array로 바꾸면 됩니다.

그렇다면 리스트를 array 형식으로 바꾸려면 어떻게 해야 하는지 아래 코드를 보면서 설명드리도록 하겠습니다.

 

먼저, 아래와 같은 리스트가 있다고 가정해보겠습니다.

 

list1 = [1,2,3,4,5]
type(list1)

 

list

 

list1의 타입은 list라고 나오는 것을 확인할 수 있습니다. 

그리고나서 이제 list1을 array로 변경하려면 다음과 같이 입력해야 합니다.

 

import numpy as np
arr = np.array(list1)
type(arr)

 

numpy.ndarray

 

위와 같이 입력한 뒤 type을 입력하면 array 형식으로 바뀌었음을 확인할 수 있습니다.

 

아래 코드는 위에서 array로 바꾼 list1을 연산하는 코드인데요,

 

print(arr + arr)

 

[ 2  4  6  8 10]

 

출력된 값을 보면, numpy 연산이 잘 실행되었음을 확인할 수 있습니다.