개발/Python

[파이썬] NumPy 배열을 파이썬 리스트로 변환

MinorMan 2023. 1. 22. 03:36
반응형

<질문>

어떻게 변환합니까?NumPy배열을 Python 목록으로?


<답변1>

사용tolist():

>>> import numpy as np
>>> np.array([[1,2,3],[4,5,6]]).tolist()
[[1, 2, 3], [4, 5, 6]]

이것은 값이 가질 수 있는 numpy 유형(예: np.int32 또는 np.float32)에서 "가장 가까운 호환 가능한 Python 유형"(목록에서)으로 값을 변환합니다. numpy 데이터 유형을 유지하려면 대신 배열에서 list()를 호출하면 됩니다.numpy scalars. (의견에서 지적해 주신 Mr_and_Mrs_D에게 감사드립니다.)


<답변2>

numpy .tolist 메서드는 numpy 배열 모양이 2D인 경우 중첩 목록을 생성합니다.

플랫 목록이 필요한 경우 아래 방법이 작동합니다.

import numpy as np
from itertools import chain

a = [1,2,3,4,5,6,7,8,9]
print type(a), len(a), a
npa = np.asarray(a)
print type(npa), npa.shape, "\n", npa
npa = npa.reshape((3, 3))
print type(npa), npa.shape, "\n", npa
a = list(chain.from_iterable(npa))
print type(a), len(a), a`

<답변3>

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

list(c.flatten())

<답변4>

tolist()중첩 배열을 만나도 잘 작동합니다. 예를 들어 pandasDataFrame;

my_list = [0,1,2,3,4,5,4,3,2,1,0]
my_dt = pd.DataFrame(my_list)
new_list = [i[0] for i in my_dt.values.tolist()]

print(type(my_list),type(my_dt),type(new_list))

<답변5>

다른 옵션

c.ravel()
# or
c.ravel().tolist()

또한 작동합니다.

반응형