Python에서 CSV 파일에 배열 쓰기

Muhammad Waiz Khan 2023년1월30일
  1. Python numpy.savetxt()메서드를 사용하여 Python에서 CSV 파일에 배열 쓰기
  2. Python에서Dataframe.to_csv()메서드를 사용하여 Python에서 CSV 파일에 배열 쓰기
  3. Python에서writer.writerows()메서드를 사용하여 Python에서 CSV 파일에 배열 쓰기
Python에서 CSV 파일에 배열 쓰기

이 자습서에서는 Python에서 배열의 데이터를 쉼표로 구분 된 값 (CSV) 파일에 쓰는 다양한 방법을 살펴 봅니다. 일부 처리 된 데이터, 일부 알고리즘의 출력, 일부 모델의 가중치 등을 포함하는 배열이 있고 나중에 사용하기 위해 저장한다고 가정합니다.

CSV 파일은 일반적으로 배열 데이터를 저장하는 데 사용됩니다. 그 형식을 사용하면 데이터를 구조화 된 테이블 형식으로 저장할 수 있습니다. Python에서 다음 방법을 사용하여 CSV 파일에 배열을 작성할 수 있습니다.

Python numpy.savetxt()메서드를 사용하여 Python에서 CSV 파일에 배열 쓰기

numpy.savetxt(fname, array, delimiter=)메소드는 입력 배열을fname파일에 저장합니다. delimiter인수는 데이터에서 독립 값을 구분하기위한 구분 기호로 사용되는 값입니다.

배열을 CSV 파일에 저장하려면.csv확장자와 함께fname을 전달하고 쉼표,를 구분 기호로 사용해야합니다. 아래 예제 코드는numpy.savetxt()를 사용하여 Python에서 CSV 파일에 배열을 작성하는 방법을 보여줍니다.

import numpy as np

a = np.array([[1, 4, 2], [7, 9, 4], [0, 6, 2]])
np.savetxt("myfile.csv", a, delimiter=",")

Python에서Dataframe.to_csv()메서드를 사용하여 Python에서 CSV 파일에 배열 쓰기

Dataframe.to_csv(path, sep,...)메소드는DataFramepath인수에 지정된 파일 및 경로에 저장합니다. sep인수는 값을 구분하는 데 사용되는 구분 기호 또는 구분 기호입니다. sep인수의 기본값은 쉼표,입니다.

먼저 배열을DataFrame으로 변환 한 다음Dataframe.to_csv()메소드를 사용하여 CSV 파일의 경로를path인수로 제공하여 CSV 파일에 배열을 작성할 수 있습니다. sep인수의 기본값이,이므로DataFramepath인수를Dataframe.to_csv()메소드에 제공해야합니다.

아래 예제 코드는Dataframe.to_csv()메서드를 사용하여 Python에서 CSV 파일에 배열을 쓰는 방법을 보여줍니다.

import pandas as pd

df = pd.DataFrame(myarray)
df.to_csv("myfile.csv")

Python에서writer.writerows()메서드를 사용하여 Python에서 CSV 파일에 배열 쓰기

writer.writerows(rows)는 2D 배열 또는 목록의 형태로rows인수를 받아writer의 파일 객체에 씁니다. writer객체는csv.writer(file)메소드에서 반환되며, 이는 파일 객체file을 입력으로 사용하고writer객체를 출력으로 반환합니다.

CSV 파일에 배열을 쓰려면 먼저 CSV 파일 객체를csv.writer()메서드에 전달한 다음 반환 된writer 객체를 사용하여 배열을 CSV 파일에 씁니다.

아래 예제 코드는writer.writerows()메서드를 사용하여 Python에서 CSV 파일에 배열을 쓰는 방법을 보여줍니다.

import csv
import numpy

a = numpy.array([[1, 4, 2], [7, 9, 4], [0, 6, 2]])

with open("myfile.csv", "w", newline="") as file:
    mywriter = csv.writer(file, delimiter=",")
    mywriter.writerows(a)

CSV 파일에서 배열을 읽으려면csv.reader(file)메소드의writer객체를 사용하여 Python에서 CSV 파일의file파일 객체에서 배열을 읽을 수 있습니다.

아래 예제 코드는csv.reader()메소드를 사용하여 CSV 파일에서 배열을 읽는 방법을 보여줍니다.

import csv

with open("myfile.csv", "r", newline="") as file:
    myreader = csv.reader(file, delimiter=",")
    for rows in myreader:
        print(rows)

출력:

['1', '4', '2']
['7', '9', '4']
['0', '6', '2']

관련 문장 - Python Array