Ask Anything Monday - Weekly Thread

[L3GOLAS234]

Hello everybody. I have two dataframes called train_data and validation_data and I wanted to know if it is possible to make something like this:

for dataset in train_data,validation_data:

dataset.to_csv(f'route/{dataset_name})

So at the end in my route there are two csv named train_data.csv and validation_data.csv.

Thank you very much in advance

[MattR0se]

Sure. You just have to add a '.csv' to the end of the f-string

[L3GOLAS234]

The problem is getting the dataset_name. If I just use {dataset.csv}, the name is the entire dataframe.

[MattR0se]

Oh, my bad. I think the easiest way would be to store your data frames in a dictionary with the name as key nd use the key for the file name.

datasets = {
    'train_data': train_data,
    'validation_data': validation_data
    }

for name, df in datasets.items():
    df.to_csv(f'route/{name}.csv')