Carl
02/23/2021, 2:41 AMChris White
Carl
02/23/2021, 9:45 AMdict
or something similar? Or perhaps an alternative way to make referencing the data in a single row more readable?
import pandas as pd
from prefect import task, Flow, Parameter
@task()
def extract_file(filename):
df = pd.read_csv(filename)
return df.values
@task()
def transform(df_row):
#do something
#print(df_row['colA'])
return df_row
@task()
def load(df):
print(df)
return
def build_flow():
with Flow('Test ETL') as flow:
df = extract_file(filename)
df_row = transform.map(df)
r = load(df_row)
return flow
flow = build_flow()
flow.run(parameters={'filename': 'data/somefile.csv'})