개발 꿀팁/PYTHON

파이썬 CSV 데이터 읽기

Jammie 2022. 11. 30. 12:22
반응형

파이썬은 두 가지 유형의 CSV 데이터를 읽습니다.
1.줄별 csv 읽기 (CSV 라이브러리 사용)

2.csv를 열별로 읽기(Pandas 라이브러리 사용)

첫 번째, 한 줄씩 읽어보도록 하겠습니다.

1.줄별 csv 읽기 (CSV 라이브러리 사용)
코드는 다음과 같습니다

import csv
if __name__ =='__main__':
    filePath="test.csv"
    with open(filePath, 'r',encoding="utf-8") as csvfile:
        reader = csv.reader(csvfile)
        for row in reader:
            #the first row is table header
            print(row)
            #type:list
            print(type(row))

첫 번째 인쇄된 row는 헤더의 내용이며, 이후 모든 내용이 한 줄씩 인쇄되며, row의 종류는 list입니다.

2.csv를 열별로 읽기(Pandas 라이브러리 사용)
test.csv 파일을 예로 들면, 파일에는 parameter와 importance라는 이름의 두 열(표두 포함)이 포함되어 있습니다.다음으로 pandas 라이브러리로 열별로 읽어 list에 각각 저장한다

코드는 다음과 같습니다

import pandas as pd
if __name__ =='__main__':
    filePath="test.csv"
    df=pd.read_csv("test.csv")
    parameterList=df["parameter"].values.tolist()
    importanceList=df["importance"].values.tolist()
    print(parameterList)
    print(importanceList)

 

반응형

'개발 꿀팁 > PYTHON' 카테고리의 다른 글

Python사전  (0) 2022.11.30
파이썬은 입문부터 마스터까지 한 달이면 충분하다  (1) 2022.11.30
pycharm 설치 pytorch  (0) 2022.11.30
파이썬 딥 카피 라이트 카피  (0) 2022.11.30
파이썬 OJ 입출력  (0) 2022.11.30