def enrich_with_geo(row): # Assume get_geo is a fast lookup function row["country"] = get_geo(row["ip"]) return row