how to create a loop and save the data array variable individually?

Question

I write a helper function read_xyT to read all .csv files in a directory and output them into a pandas DataFrame. Since there are lots for such directories and I want to save them into individual variables (or maybe some better solutions?)

what I do now is

path = r'./data/T-600s'
df33 = read_xyT(path,33)
df34 = read_xyT(path,34)
df35 = read_xyT(path,35)
df36 = read_xyT(path,36)
...

There is in total 60 folders... I wonder if there is a smarter way to do it more efficiently? i.e.

subdirectoynames = np.arange(34,50) # the helper function take 'path' as input
variablenames = alistforfilenames
for v,f in (variablenames, subdirectorynames):
  dfxx = read_xyT(path,yy)

then I'll have the saved individual variables such as df37, df38, df39, ...

Or is there a better way to do it?

Thank you in advance!

Why not dictionary? d = {'df33': read_xyT(path,33), 'df34': read_xyT(path,34)}. You can create it with a loop. — Guy
– Guy, Commented Sep 11, 2022 at 9:15
@Guy thanks for the hint, before I just know there is something called "dictionary", now I kind of understand what it is :) — Ann
– Ann, Commented Sep 11, 2022 at 13:24

code-lukas · Accepted Answer · 2022-09-11 10:55:20Z

1

You can use a dict comprehension:

df_dict = {f'df{idx}': read_xyT(path, idx) for idx in subdirectorynames}

This creates a dictionary where you can access the dataframes using e.g. df_dict['df33']

answered Sep 11, 2022 at 10:55

code-lukas

1,68113 silver badges20 bronze badges

Sign up to request clarification or add additional context in comments.

2 Comments

Ann Over a year ago

Thank you so much!! I am kind of new to python, or coding in general. I did it manually because I failed to find a solution by googling, but good that I decided to ask :) and learned a new method "dictionary" :D

code-lukas Over a year ago

I'm glad I could help! Best of luck on your journey!

Collectives™ on Stack Overflow

how to create a loop and save the data array variable individually?

1 Answer 1

2 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

2 Comments

Your Answer

Sign up or log in

Post as a guest

Related