Python에서 csv 파일을 읽고 쓰는 방법

Python에서 CSV에 쓰려면 Python의 csv 모듈을 사용하세요.

예를 들어 새 CSV 파일에 문자열 목록을 작성해 보겠습니다.

import csv
data = ["This", "is", "a", "Test"]
with open('example.csv', 'w') as file:
    writer = csv.writer(file)
    writer.writerow(data)

그러면 현재 폴더에 example.csv라는 파일이 표시됩니다.

Python에서 CSV를 작성하는 4단계

Python에서 CSV 파일을 작성하려면:

1.

쓰기 모드에서 CSV 파일을 엽니다. 이는 open() 함수를 사용하여 발생합니다. 첫 번째 인수로 파일 경로를 지정합니다. 두 번째 인수로 모드를 지정합니다(읽기의 경우 "r", 쓰기의 경우 "w"). 2.CSV 작성기 개체를 만듭니다.
이렇게 하려면 csv 모듈의writer() 객체를 생성하고 열린 파일을 인수로 전달합니다. 3. CSV 파일에 데이터를 씁니다.
CSV 파일에 데이터를 쓰려면 라이터 객체의 writerow() 함수를 사용하세요. 4. CSV 파일을 닫습니다
, ​​파일의 close() 메소드를 사용하세요.

다음은 이 프로세스를 보여주는 예입니다.

import csv
# 1.
file = open('test.csv', 'w')
# 2.
writer = csv.writer(file)
# 3.
data = ["This", "is", "a", "Test"]
writer.writerow(data)
# 4.
file.close()

이 코드는 현재 폴더에 test.csv라는 파일을 생성합니다.

지정된 파일이 존재하지 않으면 open() 함수가 새 파일을 엽니다. 그렇다면 기존 파일이 열립니다.

Sketch

CSV 작성 시간을 단축하려면

with

문을 사용하여 파일을 엽니다. 이렇게 하면 파일을 직접 닫는 것에 대해 걱정할 필요가 없습니다. with가 해당 부분을 자동으로 처리해 드립니다. 예:

import csv
# 1. step
with open('test.csv', 'w') as file:
    # 2. step
    writer = csv.writer(file)
    # 3. step
    data = ["This", "is", "a", "Test"]
    writer.writerow(data)

이렇게 하면 현재 폴더에 test.csv라는 새 CSV 파일이 생성되고 여기에 문자열 목록이 기록됩니다.

Python에서 ASCII가 아닌 문자를 CSV에 쓰는 방법

기본적으로 ASCII가 아닌 문자를 CSV 파일에 쓸 수 없습니다.

CSV 파일에 ASCII가 아닌 값 쓰기를 지원하려면 open() 호출에서 세 번째 인수로 문자 인코딩을 지정하세요.

with open('PATH_TO_FILE.csv', 'w', encoding="UTF8")

나머지 과정은 이전에 배운 단계를 따릅니다.

CSV 파일의 헤더를 만드는 방법

지금까지 구조가 부족한 CSV 파일을 만들었습니다.

Python에서는

writerow() 함수를 사용하여 모든 CSV 파일에 대한 헤더를 작성할 수 있습니다.

예: 학생 데이터가 포함된 샘플 CSV 파일을 만들어 보겠습니다.

데이터를 효율적으로 구축하려면 CSV 파일 시작 부분에 학생용 헤더를 생성하고 삽입해야 합니다. 이전과 동일한 단계를 수행하여 데이터를 CSV 파일에 쓰면 작업이 완료됩니다.

코드는 다음과 같습니다:

import csv
# Define the structure of the data
student_header = ['name', 'age', 'major', 'minor']
# Define the actual data
student_data = ['Jack', 23, 'Physics', 'Chemistry']
# 1. Open a new CSV file
with open('students.csv', 'w') as file:
    # 2. Create a CSV writer
    writer = csv.writer(file)
    # 3. Write data to the file
    writer.writerow(student_header)
    writer.writerow(student_data)

이렇게 하면 현재 작업 중인 폴더에 Students.csv 파일이 생성됩니다. 새 파일은 다음과 같습니다.

Python에서 CSV 파일에 여러 행을 쓰는 방법

Python에서는 CSV 작성기의 writerows() 함수를 사용하여 CSV 파일에 여러 행을 동시에 쓸 수 있습니다.

예. 여러 행의 데이터를 CSV 파일에 쓰려고 한다고 가정해 보겠습니다. 예를 들어, 단 한 명의 학생이 아닌 학생 목록이 있을 수 있습니다.

CSV에 여러 행의 데이터를 쓰려면 writerows() 메서드를 사용하세요.

예는 다음과 같습니다.

import csv
student_header = ['name', 'age', 'major', 'minor']
student_data = [
    ['Jack', 23, 'Physics', 'Chemistry'],
    ['Sophie', 22, 'Physics', 'Computer Science'],
    ['John', 24, 'Mathematics', 'Physics'],
    ['Jane', 30, 'Chemistry', 'Physics']
]
with open('students.csv', 'w') as file:
    writer = csv.writer(file)
    writer.writerow(student_header)
    # Use writerows() not writerow()
    writer.writerows(student_data)

이렇게 하면 아래와 같이 새 CSV 파일이 생성됩니다.

Python에서 CSV 파일에 사전을 작성하는 방법

Python에서 사전 작성은 DictWriter를 사용하여 수행할 수 있습니다. object CSV 파일 작성에는 다음 세 단계가 포함됩니다.

1. csv 모듈의 DictWriter 개체를 사용하고 그 안에 필드 이름을 지정합니다.

2. writeheader() 메서드를 사용하여 CSV 파일에 헤더를 만듭니다.

3. writerows() 메서드를 사용하여 사전 데이터를 파일에 씁니다.

예: 학생 데이터 사전을 CSV 파일에 작성해 보겠습니다.

import csv
student_header = ['name', 'age', 'major', 'minor']
student_data = [
    {'name': 'Jack', 'age': 23, 'major': 'Physics', 'minor': 'Chemistry'},
    {'name': 'Sophie', 'age': 22, 'major': 'Physics', 'minor': 'Computer Science'},
    {'name': 'John', 'age': 24, 'major': 'Mathematics', 'minor': 'Physics'},
    {'name': 'Jane', 'age': 30, 'major': 'Chemistry', 'minor': 'Physics'}
]
with open('students.csv', 'w') as file:
    # Create a CSV dictionary writer and add the student header as field names
    writer = csv.DictWriter(file, fieldnames=student_header)
    # Use writerows() not writerow()
    writer.writeheader()
    writer.writerows(student_data)

이제 결과는 이전 예의 Students.csv 파일과 동일합니다.

Conclusion

CSV 또는 Comma Separated Values ​​​​는 일반적으로 사용되는 파일 형식입니다. 일반적으로 쉼표로 구분된 값으로 구성됩니다.

Python에서 CSV를 작성하려면 다음 단계에 따라 csv 모듈을 사용해야 합니다.

1. 쓰기 모드에서 CSV 파일을 엽니다.

2. CSV 작성기 개체를 만듭니다.

3. CSV 파일에 데이터를 씁니다.

4.CSV 파일을 닫습니다.

실용적인 예입니다.

import csv
data = ["This", "is", "a", "Test"]
with open('example.csv', 'w') as file:
    writer = csv.writer(file)
    writer.writerow(data)

즐거운 코딩하세요!

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