brian_gtbB
Windmill17mo ago
3 replies
brian_gtb

[Solved] DataFrame in Script and Flow Showing Incorrect Headers

I always got confused why my DataFrame header is not correct. You may see this sample code, I printed the logs also to see what Column Header it should have vs what being shown in Windmill. Please see the image. Let me know if the issue is on my side or Windmill's.

I expect the column headers: ID, Amount, Status
What being shown in Windmill: col0, col1, col2

Thanks.

import pandas as pd
from typing import Any
import csv
import io

def main() -> Any:
    # Create a dummy CSV data string with 3 columns
    csv_data = """ID;Amount;Status
    abc123;1000;completed
    def456;2000;pending
    ghi789;3000;failed"""

    # Use StringIO to treat the string as a file object
    csv_file_like_object = io.StringIO(csv_data)

    # Read the CSV data into a DataFrame directly
    df = pd.read_csv(csv_file_like_object, delimiter=';')

    # Debugging: Print DataFrame information
    print("DataFrame columns:", df.columns)
    print("DataFrame head:\n", df.head())
    
    # Return the DataFrame
    return df
image.png
Was this page helpful?