파이썬으로 데이터를 처리하다 보면 Pandas와 Numpy를 많이 사용하게 됩니다. 이번 포스팅에서는 판다스 데이터프레임을 넘파이로 간단하게 변환하는 두 가지 방법에 대해 여러분과 공유하고자 합니다.
최근 파이썬을 활용한 프로젝트를 수행하고 있으며 자주 사용하는 코드를 저장하기 위해 포스팅을 연재하고 있습니다. 문의 사항은 댓글에 남겨주시면 되겠습니다.
예제 적용 데이터프레임
예제에 적용하기 위해 Pandas dataframe이 필요합니다. 해당 데이터는 공식 Pandas API 레퍼런스에서 발최했습니다. 해당 데이터는 아래와 같습니다. 이는 넘파이를 판다스로 변환시키는 예제로 활용할 수도 있습니다. 해당 코드는 아래와 같습니다.
import pandas as pd
import numpy as np
df = pd.DataFrame(np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]]),
columns=['a', 'b', 'c'])
Pandat to Numpy 변환 예제
판다스 데이터프레임을 넘파이로 변환하는 두 가지 방법에 대해 알아보겠습니다. 간단한 한 줄짜리 코드로 변환할 수 있습니다.
방법1. values
위에서 설명드린 바와 같이 판다스 데이터 프레임을 넘파이로 변환하는 데는 두 가지 방법이 존재합니다. 제가 가장 많이 사용하는 방법은 values 매소드를 이용한 방법입니다. 아래 예시와 같이 적용 가능합니다.
df.values
방법2. to_numpy()
판다스 데이터프레임을 넘파이로 변환하는 두 번째 방법은 to_numpy() 매소드를 사용하는 방법입니다. 이를 사용하면 values 매소드를 사용한 것과 동일한 결과값을 도출할 수 있습니다.
df.to_numpy()
예제 결과
위와 같이 방법1, 방법2를 적용한 결과는 아래와 같습니다. 동일한 결과를 출력하는 것을 확인해볼 수 있습니다.
array([[1, 2, 3],
[4, 5, 6],
[7, 8, 9]])
마치며
이번 포스팅에서는 간단히 Pandas dataframe을 Numpy array로 변환하는 방법을 예제를 통해 알아봤습니다. 해당 내용 참고하시고 업무와 학업에 도움이 되었으면 좋겠습니다.
댓글