enrich = lambda src: src.map(enrich_with_geo) Now enrich can be inserted anywhere in a pipeline:
def enrich_with_geo(row): # Assume get_geo is a fast lookup function row["country"] = get_geo(row["ip"]) return row
def capitalize_name(row): row["name"] = row["name"].title() return row