brian_gtbB
Windmill17mo ago
11 replies
brian_gtb

DataFrame in Script and Flow Showing Incorrect Column Order

I created a dataframe firstly with these 3 columns in this order: ["ID", "Amount", "Status"]
Then I reorder the column to ["Status", "ID", "Amount"]

However when shown in Windmill, it doesn't get reordered.

This is just a simplification of my code, in reality I'm working with more complex column management, but this issue makes me cannot proceed.

Here is the full code
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=';')
    df2 = df[['Status', 'ID', 'Amount']]

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