postgresql에 저장된 데이터를 csv로 바꾸는 방법

joon·2023년 1월 8일
0

Postgre를 내보내려면SQL 데이터베이스에서 CSV 파일로, psycopg2 라이브러리를 사용하여 데이터베이스에 연결하고 csv 라이브러리를 사용하여 데이터를 CSV 파일에 쓸 수 있습니다. 다음은 이 작업을 수행하는 방법의 예입니다:

import psycopg2
import csv

# Connect to the database
conn = psycopg2.connect(database="mydatabase", user="myuser", password="mypassword", host="localhost", port="5432")

# Open a cursor to perform database operations
cur = conn.cursor()

# Execute a statement to get the data you want to export
cur.execute("SELECT * FROM mytable")

# Fetch all the rows of the result set
rows = cur.fetchall()

# Open a file for writing
with open('mytable.csv', 'w', newline='') as f:
  # Create a CSV writer
  writer = csv.writer(f)

  # Write the header row
  writer.writerow([d.name for d in cur.description])

  # Write all the rows
  for row in rows:
      writer.writerow(row)

# Close the file and the database connection
f.close()
conn.close()

이 코드는 로컬 호스트의 "mydatabase"라는 데이터베이스에 연결하고, SELECT 문을 실행하여 "mytable"이라는 테이블에서 모든 행을 가져오고, 행을 "mytable.csv"라는 CSV 파일에 씁니다. 또한 코드는 테이블의 열 이름을 포함하는 헤더 행을 작성합니다.

0개의 댓글