Pandas 시리즈 또는 인덱스를 NumPy 배열로 변환하는 방법

2024-07-27

to_numpy() 메서드 사용:

Pandas 0.24.0 버전부터는 to_numpy() 메서드를 사용하여 시리즈 또는 인덱스를 쉽게 NumPy 배열로 변환할 수 있습니다. 이 메서드는 다음과 같이 사용됩니다.

import pandas as pd
import numpy as np

# 시리즈 만들기
s = pd.Series([1, 2, 3, 4, 5])

# 시리즈를 NumPy 배열로 변환
array = s.to_numpy()

# 결과 출력
print(array)

위 코드는 다음과 같은 결과를 출력합니다.

[1 2 3 4 5]

to_numpy() 메서드는 인덱스도 NumPy 배열로 변환하는 데 사용할 수 있습니다.

# 인덱스 만들기
index = pd.Index([10, 20, 30, 40, 50])

# 인덱스를 NumPy 배열로 변환
array = index.to_numpy()

# 결과 출력
print(array)

[10 20 30 40 50]

np.array() 함수 사용:

np.array() 함수를 사용하여 Pandas 시리즈 또는 인덱스를 NumPy 배열로 변환할 수도 있습니다. 이 방법은 다음과 같이 사용됩니다.

import pandas as pd
import numpy as np

# 시리즈 만들기
s = pd.Series([1, 2, 3, 4, 5])

# 시리즈를 NumPy 배열로 변환
array = np.array(s)

# 결과 출력
print(array)

위 코드는 to_numpy() 메서드를 사용하는 경우와 동일한 결과를 출력합니다.

values 속성 사용:

Pandas 시리즈의 values 속성을 사용하여 NumPy 배열을 얻을 수도 있습니다. 하지만 이 속성은 시리즈의 데이터만 반환하고 인덱스는 반환하지 않는다는 점에 유의해야 합니다.

import pandas as pd

# 시리즈 만들기
s = pd.Series([1, 2, 3, 4, 5])

# NumPy 배열 얻기
array = s.values

# 결과 출력
print(array)

[1 2 3 4 5]

결론

Pandas 시리즈 또는 인덱스를 NumPy 배열로 변환하는 방법에는 여러 가지가 있습니다. to_numpy() 메서드는 Pandas 0.24.0 버전부터 도입된 새 메서드로 가장 간편하고 효율적인 방법입니다.

예제 코드

to_numpy() 메서드 사용

import pandas as pd
import numpy as np

# 시리즈 만들기
s = pd.Series([1, 2, 3, 4, 5])

# 시리즈를 NumPy 배열로 변환
array = s.to_numpy()

# 결과 출력
print(array)

np.array() 함수 사용

import pandas as pd
import numpy as np

# 시리즈 만들기
s = pd.Series([1, 2, 3, 4, 5])

# 시리즈를 NumPy 배열로 변환
array = np.array(s)

# 결과 출력
print(array)

values 속성 사용

import pandas as pd

# 시리즈 만들기
s = pd.Series([1, 2, 3, 4, 5])

# NumPy 배열 얻기
array = s.values

# 결과 출력
print(array)

추가 예제

다음은 Pandas 데이터프레임의 열을 NumPy 배열로 변환하는 방법을 보여주는 예제 코드입니다.

import pandas as pd
import numpy as np

# 데이터프레임 만들기
df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6], 'C': [7, 8, 9]})

# 특정 열을 NumPy 배열로 변환
array = df['A'].to_numpy()

# 결과 출력
print(array)

[1 2 3]

Pandas 시리즈 또는 인덱스를 NumPy 배열로 변환하는 대체 방법

astype() 메서드 사용:

astype() 메서드를 사용하여 Pandas 시리즈 또는 인덱스를 numpy.ndarray 데이터 타입으로 변환할 수 있습니다. 이 방법은 다음과 같이 사용됩니다.

import pandas as pd
import numpy as np

# 시리즈 만들기
s = pd.Series([1, 2, 3, 4, 5])

# 시리즈를 NumPy 배열로 변환
array = s.astype(np.ndarray)

# 결과 출력
print(array)

[1 2 3 4 5]

tolist() 메서드 사용:

tolist() 메서드를 사용하여 Pandas 시리즈 또는 인덱스를 Python 리스트로 변환한 다음, np.array() 함수를 사용하여 NumPy 배열로 변환할 수 있습니다. 이 방법은 다음과 같이 사용됩니다.

import pandas as pd
import numpy as np

# 시리즈 만들기
s = pd.Series([1, 2, 3, 4, 5])

# 시리즈를 리스트로 변환
list = s.tolist()

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

# 결과 출력
print(array)

위 코드는 앞서 설명한 예제 코드들과 동일한 결과를 출력합니다.

직접 for 루프 사용:

import pandas as pd
import numpy as np

# 시리즈 만들기
s = pd.Series([1, 2, 3, 4, 5])

# NumPy 배열 만들기
array = np.array([])

# for 루프를 사용하여 시리즈의 각 요소를 배열에 추가
for item in s:
    array = np.append(array, item)

# 결과 출력
print(array)

결론

Pandas 시리즈 또는 인덱스를 NumPy 배열로 변환하는 방법에는 여러 가지가 있습니다. 상황에 따라 가장 적합한 방법을 선택하면 됩니다.

to_numpy() 메서드는 Pandas 0.24.0 버전부터 도입된 새 메서드로 가장 간편하고 효율적인 방법입니다.
np.array() 함수는 Pandas 시리즈 또는 인덱스를 직접 NumPy 배열로 변환하는 데 사용할 수 있습니다.
values 속성은 Pandas 시리즈의 데이터만 NumPy 배열로 변환하는 데 사용할 수 있습니다.
astype() 메서드는 Pandas 시리즈 또는 인덱스를 numpy.ndarray 데이터 타입으로 변환하는 데 사용할 수 있습니다.
tolist() 메서드와 np.array() 함수를 함께 사용하여 Pandas 시리즈 또는 인덱스를 NumPy 배열로 변환할 수 있습니다.

python pandas