>  기사  >  백엔드 개발  >  Python 3.x에서 csv 모듈을 사용하여 CSV 파일을 읽고 쓰는 방법

Python 3.x에서 csv 모듈을 사용하여 CSV 파일을 읽고 쓰는 방법

王林
王林원래의
2023-07-29 23:49:121767검색

csv 모듈을 사용하여 Python 3.x에서 CSV 파일을 읽고 쓰는 방법

소개:
CSV(쉼표로 구분된 값)는 표 형식 데이터를 저장하는 데 사용되는 일반적인 파일 형식입니다. Python에서는 CSV 파일을 읽고 쓰는 기능을 제공하는 csv 모듈을 사용하여 CSV 파일을 처리할 수 있습니다. 이 문서에서는 csv 모듈을 사용하여 Python 3.x에서 CSV 파일을 읽고 쓰는 방법을 설명하고 코드 예제를 제공합니다.

1. csv 모듈을 사용하여 CSV 파일을 읽습니다

먼저 csv 모듈을 가져와야 합니다.

import csv

다음으로 csv.reader() 함수를 사용하여 CSV 파일을 읽을 수 있습니다. 이 함수는 파일 객체를 전달해야 합니다. open() 함수를 사용하여 CSV 파일을 열고 파일 객체를 csv.reader() 함수에 매개변수로 전달할 수 있습니다.

다음 내용이 포함된 data.csv라는 CSV 파일이 있다고 가정합니다.

name,age,city
John,25,New York
Peter,30,Chicago
Sarah,28,Los Angeles

다음은 csv 모듈을 사용하여 CSV 파일을 읽는 방법을 보여주는 간단한 코드 예제입니다.

import csv

with open('data.csv', 'r') as file:
    csv_reader = csv.reader(file)
    for row in csv_reader:
        print(row)

위 코드를 실행합니다. CSV 파일의 데이터가 출력됩니다. 이 예에서는 for 루프를 사용하여 csv_reader 개체를 반복하여 CSV 파일의 데이터를 한 줄씩 인쇄합니다.

2. csv 모듈을 사용하여 CSV 파일 작성

CSV 파일을 읽는 것과 마찬가지로 csv 모듈을 사용하여 CSV 파일을 작성할 수도 있습니다. 또한 csv 모듈을 가져와야 합니다:

import csv

그런 다음 csv.writer() 함수를 사용하여 CSV 파일에 데이터를 쓸 수 있는 csv_writer 개체를 만들 수 있습니다. 이 함수는 파일 객체를 전달해야 합니다. open() 함수를 사용하여 CSV 파일을 열고 파일 객체를 csv.writer() 함수에 매개 변수로 전달할 수도 있습니다.

다음은 csv 모듈을 사용하여 CSV 파일을 작성하는 방법을 보여주는 간단한 코드 예제입니다.

import csv

data = [
    ['name', 'age', 'city'],
    ['John', '25', 'New York'],
    ['Peter', '30', 'Chicago'],
    ['Sarah', '28', 'Los Angeles']
]

with open('output.csv', 'w', newline='') as file:
    csv_writer = csv.writer(file)
    csv_writer.writerows(data)

위 코드를 실행하면 현재 디렉터리에 output.csv라는 이름의 CSV 파일이 생성되고 Data에 데이터가 생성됩니다. 이 파일에 기록됩니다.

코드 예제에서는 먼저 CSV 파일의 각 데이터 행을 포함하는 2차원 목록 데이터를 정의합니다. 그런 다음 csv.writerows() 함수를 사용하여 전체 목록을 CSV 파일에 씁니다.

요약:
csv 모듈을 사용하면 Python 3.x에서 CSV 파일을 쉽게 읽고 쓸 수 있습니다. CSV 파일을 읽을 때 csv.reader() 함수를 사용하고 for 루프를 사용하여 데이터를 한 줄씩 처리할 수 있습니다. CSV 파일을 쓸 때는 csv.writer() 함수와 csv_writer.writerows() 함수를 사용할 수 있습니다. 데이터는 CSV 파일에 기록됩니다.

이 기사가 Python의 csv 모듈을 사용하여 CSV 파일을 읽고 쓰는 데 도움이 되기를 바랍니다. 즐겁게 공부하세요!

위 내용은 Python 3.x에서 csv 모듈을 사용하여 CSV 파일을 읽고 쓰는 방법의 상세 내용입니다. 자세한 내용은 PHP 중국어 웹사이트의 기타 관련 기사를 참조하세요!

성명:
본 글의 내용은 네티즌들의 자발적인 기여로 작성되었으며, 저작권은 원저작자에게 있습니다. 본 사이트는 이에 상응하는 법적 책임을 지지 않습니다. 표절이나 침해가 의심되는 콘텐츠를 발견한 경우 admin@php.cn으로 문의하세요.