파이썬[Python] Numpy array 데이터 CSV로 추출하기

이번 포스팅에서는 Numpy데이터을 외부 추출하는 방법에 대해서 살펴볼려고 합니다.  물론 Pandas등을 통해서 CSV 파일로 추출하는 방법들을 매우 대중화 되어 있습니다. 그래서 많은 분들은 일정한 단계를 걸쳐 Pandas를 이용한 CSV파일로 추출을 합니다. 저는 바로 Pandas DataFrame으로 변환하여 CSV파일을 추출하는 방법과 바로 Numpy에서 CSV 파일을 추출하는 방법에 대해서 살펴보겠습니다. 

 

Pandas DataFrame으로 변환하여 CSV파일 추출하기 

실제 Numpy array를 Dataframe으로 변환하여 CSV파일을 추출하기 위해서는 다음 단계를 걸쳐야 합니다. 

 

1. numpy array을 pandas DataFrame으로 변경

2. CSV 파일로 저장 

 

그럼 다음 예시를 통해서 데이터를 추출하는 방법에 대해서 살펴보도록 하겠습니다. 

example) 

import pandas as pd
 
import numpy as np
 
#numpy array 생성
 
arr = np.array([[1,2,3],[3,4,5],[4,5,6]])
 
#numpy array을 pandas dataframe으로 변환
 
df = pd.DataFrame(arr)
 
df.to_csv('sample.csv', index=False)

result) 

pandas에서 csv 파일 추출 결과 

위에서 보이는 바와 같이 "to_csv()"함수를 활용하여 추출이 가능합니다. 실제 매우 간단하게 추출이 가능합니다. 그렇지만 pandas를 사용하지 않은 사람들이 있을 수 있습니다. 따라서 numpy만 활용하여 csv파일을 추출하는 방법이 있습니다. numpy에서 위의 단계 없이도 바로 CSV 파일로 추출이 가능합니다. 그 방법에 대해서 한번 살펴보도록 하겠습니다. 

 

Numpy에서 바로 추출하기 

example)

import numpy as np
 
#numpy array 생성
 
arr = np.array([[1,2,3],[3,4,5],[4,5,6]])
 
#CSV파일 추출
 
np.savetxt('sample1.csv',arr,delimiter=",")

result)

numpy에서 savetxt()를 이용하여 csv파일을 추출할 수 있습니다. 이 부분이 실제로는 txt 파일을 추출하는 방식이나 delimiter를 이용하여 csv파일을 추출할 수 있습니다. 

 

오늘은 Numpy Array에서 CSV파일을 추출하는 방법에 대해서 2가지 형태로 살펴봤습니다. 물론, csv 모듈을 활용하는 경우도 있습니다. 하지만, 그런 방법보다 이 방법이 보다 효율적이라 다음과 같이 작업을 진행해봤습니다. 혹 궁금하신 점이나 문의 사항이 있으시면 댓글 및 방명록에 글 남겨주시길 바랍니다. 감사합니다.