Python에서 CSV에 목록을 작성하는 방법

Azaz Farooq 2023년1월30일
  1. csv.writer()메서드를 사용하여 Python에서 CSV에 목록 쓰기
  2. Quotes 메서드를 사용하여 Python에서 CSV에 목록 쓰기
  3. pandas 메서드를 사용하여 Python에서 CSV에 목록 쓰기
  4. NumPy 메서드를 사용하여 Python에서 CSV에 목록 쓰기
Python에서 CSV에 목록을 작성하는 방법

CSV 파일은 데이터를 표 형식으로 저장할 수 있습니다. 이 데이터는 간단한 텍스트입니다. 모든 데이터 행을 레코드라고하며 각 행은 쉼표로 구분할 수 있습니다.

이 기사에서는 Python에서 CSV 목록을 작성하는 다양한 방법에 대해 설명합니다.

csv.writer()메서드를 사용하여 Python에서 CSV에 목록 쓰기

먼저csv 모듈을 가져 오겠습니다.

import csv

CSV 파일에 쓰기 위해 다음 항목을 사용한다고 가정합니다.

RollNo, Name, Subject
1, ABC, Economics
2, TUV, Arts
3, XYZ, Python

이제 전체 예제 코드는 다음과 같습니다.

import csv

with open("students.csv", "w", newline="") as student_file:
    writer = csv.writer(student_file)
    writer.writerow(["RollNo", "Name", "Subject"])
    writer.writerow([1, "ABC", "Economics"])
    writer.writerow([2, "TUV", "Arts"])
    writer.writerow([3, "XYZ", "Python"])

위의 코드를 실행하면students.csv 파일이 코드에있는 항목으로 현재 디렉토리에 행 단위로 생성됩니다. csv.writer()함수는writer()객체를 생성하고writer.writerow()명령은 항목을 CSV 파일에 행 단위로 삽입합니다.

Quotes 메서드를 사용하여 Python에서 CSV에 목록 쓰기

이 방법에서는 따옴표를 사용하여 CSV 파일에 값을 쓰는 방법을 알아 봅니다. 전체 예제 코드는 다음과 같습니다.

import csv

list = [["RN", "Name", "GRADES"], [1, "ABC", "A"], [2, "TUV", "B"], [3, "XYZ", "C"]]
with open("studentgrades.csv", "w", newline="") as file:
    writer = csv.writer(file, quoting=csv.QUOTE_ALL, delimiter=";")
    writer.writerows(list)

studentgrades.csv 파일이 현재 디렉토리에 생성됩니다. csv.QUOTE_ALL()함수는 구분 기호;를 사용하여 모든 항목에 큰 따옴표를 넣습니다.

출력:

RN;"Name";"GRADES"
1;"ABC";"A"
2;"TUV";"B"
3;"XYZ";"C"

pandas 메서드를 사용하여 Python에서 CSV에 목록 쓰기

이 방법은 완전한 데이터 프레임을 보유하는Pandas 라이브러리를 사용합니다. 이 라이브러리가 PC에 설치되어 있지 않은 경우 Google Colab이라는 온라인 도구를 사용할 수 있습니다.

전체 예제 코드는 다음과 같습니다.

import pandas as pd

name = ["ABC", "TUV", "XYZ", "PQR"]
degree = ["BBA", "MBA", "BSC", "MSC"]
score = [98, 90, 88, 95]
dict = {"name": name, "degree": degree, "score": score}
df = pd.DataFrame(dict)
df.to_csv("test.csv")

출력:

	name	degree	score
0	ABC	    BBA	     98
1	TUV	    MBA	     90
2	XYZ	    BSC	     88
3	PQR	    MSC	     95

이 메서드는 Python 목록을 데이터 프레임으로 CSV 파일에 씁니다.

NumPy 메서드를 사용하여 Python에서 CSV에 목록 쓰기

이 방법은NumPy 라이브러리를 사용하여 목록을 Python의 CSV 파일에 저장합니다. 전체 예제 코드는 다음과 같습니다.

import numpy as np

list_rows = [
    ["ABC", "COE", "2", "9.0"],
    ["TUV", "COE", "2", "9.1"],
    ["XYZ", "IT", "2", "9.3"],
    ["PQR", "SE", "1", "9.5"],
]
np.savetxt("numpy_test.csv", list_rows, delimiter=",", fmt="% s")

NumPy 라이브러리의savetxt()함수는 Python 목록에서 CSV 파일을 작성합니다.

출력:

ABC	 COE	2	9
TUV	 COE	2	9.1
XYZ	 IT	2	9.3
PQR	 SE	1	9.5

관련 문장 - Python CSV

관련 문장 - Python List