pandas dataframe 결과 파일 저장 및 보고서 생성

판다스 라이브러리를 사용하여 결과 파일을 저장하고 보고서를 생성하는 다양한 케이스에 대한 예시 코드와 함께 설명하겠습니다.

Case 1: 결과 파일 저장

판다스를 사용하여 데이터프레임의 결과를 CSV 파일로 저장하는 경우입니다.

import pandas as pd

# 샘플 데이터프레임 생성
data = {'이름': ['Alice', 'Bob', 'Charlie'],
        '나이': [30, 25, 35]}

df = pd.DataFrame(data)

# 데이터프레임을 CSV 파일로 저장
df.to_csv('결과파일.csv', index=False)

Case 2: 결과 파일 저장 (Excel 형식)

데이터프레임의 결과를 Excel 파일로 저장하는 경우입니다.

import pandas as pd

# 샘플 데이터프레임 생성
data = {'이름': ['Alice', 'Bob', 'Charlie'],
        '나이': [30, 25, 35]}

df = pd.DataFrame(data)

# 데이터프레임을 Excel 파일로 저장
df.to_excel('결과파일.xlsx', index=False)

Case 3: 보고서 생성

판다스와 다른 라이브러리를 함께 사용하여 PDF 형식의 보고서를 생성하는 경우입니다. 이를 위해 먼저 matplotlib를 사용하여 그래프를 생성하고, 그래프를 이미지로 저장한 다음, reportlab을 사용하여 이미지를 PDF에 삽입할 수 있습니다.

import pandas as pd
import matplotlib.pyplot as plt
from reportlab.lib.pagesizes import letter
from reportlab.pdfgen import canvas

# 샘플 데이터프레임 생성
data = {'과목': ['수학', '과학', '영어', '역사'],
        '점수': [90, 85, 78, 92]}

df = pd.DataFrame(data)

# 그래프 생성
plt.bar(df['과목'], df['점수'])
plt.title('과목별 점수')
plt.xlabel('과목')
plt.ylabel('점수')

# 그래프 이미지로 저장
plt.savefig('그래프.png')

# PDF 생성
c = canvas.Canvas('보고서.pdf', pagesize=letter)
c.drawString(100, 750, '보고서 제목')

# 그래프 이미지 삽입
c.drawImage('그래프.png', 100, 400, width=400, height=300)

c.save()

위의 예시 코드에서는 판다스를 사용하여 데이터를 처리하고, matplotlib로 그래프를 생성하고 이미지로 저장한 다음, reportlab을 사용하여 PDF 보고서를 생성하는 방법을 설명하였습니다. 이를 통해 데이터 결과를 다양한 형식으로 저장하고 보고서를 생성할 수 있습니다.