ETL 및 ELT 파이프라인 구축

Python으로 ETL과 ELT

Jake Roach

Data Engineer

CSV 파일에서 데이터 추출

import pandas as pd

# Read in the CSV file to a DataFrame
data_frame = pd.read_csv("raw_data.csv")

# Output the first few rows
data_frame.head()

   name         num_firms   total_income
0  Advertising     58         3892.41
1  Apparel         39         5422.69
...
49 Trucking        35         17324.36

read_csv()

파일 경로를 받아 DataFrame 반환
delimiter, header, engine

.head()

DataFrame의 처음 n개 행 출력

DataFrame 필터링

   name         num_firms   total_income
0  Advertising     58         3892.41
1  Apparel         39         5422.69
...
49 Trucking        35         17324.36


         name         num_firms
      1  Apparel         39
      37 Apparel         61

# First, by rows
data_frame.loc[data_frame["name"] == "Apparel", :]

# Then, by columns
data_frame.loc[:, ["name", "num_firms"]]

.loc

DataFrame 필터링
: 는 "전체"를 의미

DataFrame을 CSV로 저장

# Write a DataFrame to a .csv file
data_frame.to_csv("cleaned_data.csv")

.to_csv()

path를 받아 해당 위치에 파일로 저장
출력 형식을 조정하는 추가 매개변수 지원

기타 옵션:

.to_json(), .to_excel(), .to_sql()

SQL 쿼리 실행

data_warehouse.execute(  # Use Python clients or other tools to run SQL queries
    """
    CREATE TABLE total_sales AS 
        SELECT
            ds,
            SUM(sales)
        FROM raw_sales_data
        GROUP BY ds;
    """
)

.execute() 등으로 SQL 쿼리 실행

모두 합쳐서 실행하기!

# Define extract(), transform(), and load() functions
...

def transform(data_frame, value):
    return data_frame.loc[data_frame["name"] == value, ["name", "num_firms"]]

# First, extract data from a .csv
extracted_data = extract(file_name="raw_data.csv")

# Then, transform the `extracted_data`
transformed_data = transform(data_frame=extracted_data, value="Apparel")

# Finally, load the `transformed_data`
load(data_frame=transformed_data, file_name="cleaned_data.csv")

Let's practice!

Python으로 ETL과 ELT