BLOG/파이썬 기초
[파이썬] 리스트를 numpy 배열(array)로 바꾸려면
알디노
2021. 3. 8. 16:05
파이썬에서 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 연산이 잘 실행되었음을 확인할 수 있습니다.