본문 바로가기

BLOG/파이썬 기초

[파이썬] 주피터 노트북에서 csv, 엑셀 파일 열기 - open(), reader()

 

데이터프레임 등으로 데이터 분석을 할 때, 주로 csv 파일 또는 엑셀 형식의 파일을 불러와 데이터 분석을 하게 되는데요, 이때 csv 파일과 엑셀 파일을 어떻게 불러오는지 알아보도록 하겠습니다.

 

 

1. csv 파일 열기

먼저, csv 파일을 불러오려면 csv 라이브러리를 import 해야 합니다. open() 함수를 이용하여 파일을 열고 'f' 라는 변수에 저장하도록 하겠습니다.

 

import csv
f = open(r"C\test\sample.csv", encoding="utf-8") #파일이 있는 경로+파일이름.csv

 

그리고나서 csv.reader() 함수를 이용하여 csv 파일을 읽도록 합니다.

 

f_csv = csv.reader(f)

 

이렇게 코드를 입력하면 f_csv 에는 csv 파일에 있는 데이터가 들어가게 됩니다.

 

 

2. csv 파일 읽기

한편, 불러온 csv 파일을 읽는 방법은 다양한데요, 그 중 간단하게 읽을 수 있는 코드는 다음과 같습니다.

 

for i in f_csv :
    print(i)

 

위와 같이 입력하면 csv 파일에 있는 내용이 잘 출력됨을 확인할 수 있습니다.

 

 

3. excel 파일 열기

반면 excel 파일도 위와 동일하게 불러오면 되는데요, 다른 점이 있다면 여기서는 openpyxl 이라는 라이브러리를 사용합니다.

 

import openpyxl
wb = openpyxl.load_workbook(r"C\test\sample.xlsx")
sheet = wb['Sheet1']

 

위와 같이 코드를 입력하면 엑셀 파일을 열게 되고, 이때 엑셀 파일을 열 때는 sheet의 이름도 설정해주어야 한답니다.