파이썬 List를 Numpy 배열로 변환

파이썬의 list를 Numpy 배열로 변환하는 방법을 소개합니다.

1차원 리스트를 Numpy 배열로 변환

아래와 같이 np.array(list)는 리스트를 numpy 배열로 변환합니다.

import numpy as np

# 리스트 생성
lst = [1, 2, 3, 4, 5]

# 리스트를 배열로 변환
arr = np.array(lst)

print(arr)

Output:

[1 2 3 4 5]

2차원 리스트를 Numpy 배열로 변환

아래와 같이 np.array(list)는 2차원 리스트를 2차원의 numpy 배열로 변환합니다.

import numpy as np

# 리스트 생성
lst = [[1, 2, 3], [4, 5, 6]]

# 리스트를 배열로 변환
arr = np.array(lst)

print(arr)

Output:

[[1 2 3] [4 5 6]]

다른 타입 리스트를 Numpy 배열로 변환

다양한 타입이 섞여 있는 리스트를 Numpy 배열로 변환할 수도 있습니다.

numpy.array()로 리스트를 배열로 변환할 때, 데이터 타입(dtype)을 설정하여 int나 float으로 변환할 수 있습니다.

import numpy as np

# 리스트 생성
lst = ['1', '2', 3, 4.0, 5.0]

# 리스트를 int 배열로 변환
arr = np.array(lst, dtype=int)
print(arr)

# 리스트를 float 배열로 변환
arr = np.array(lst, dtype=float)
print(arr)

Output:

[1 2 3 4 5]
[1. 2. 3. 4. 5.]
Loading script...

Related Posts

codechachaCopyright ©2019 codechacha