Extract data from Pandas dataframe

Question

I created a dataframe (called df) with few columns (for example 3 columns with the following names: 'data', 'SID' and 'time'). df has only one row. Each cell of the data frame contains a dictionary built as follow:

{'column_name': ndarray}

For example, the dictionary at the SID column is like this:

{'SID': ndarray}

I would like to access/extract the ndarrays stored in the dictionaries and assign each one of them to a new variable called after the column name as follows:

data = df.at[0, 'data']['data']
SID = df.at[0, 'SID']['SID']
time = df.at[0, 'time']['time']

I'm asking for a generic/iterative way to do it. Any idea?

pyigal · Accepted Answer · 2017-06-16 09:02:48Z

1

Your approach seems a bit 'hacky', but you can use:

   for column in df:
     exec(column + '= df.at[0, column][column]')

edited Jun 16, 2017 at 9:02

pyigal

3891 gold badge5 silver badges16 bronze badges

answered Jun 13, 2017 at 12:27

SeaMonkey

1319 bronze badges

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

2 Comments

pyigal Over a year ago

The created variable is always called column, therefore, in each iteration, the variable is being overridden.

SeaMonkey Over a year ago

Indeed, my bad. I changed it

Collectives™ on Stack Overflow

Extract data from Pandas dataframe

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