Create dict from a pandas dataframe with values as tuple

Question

I have a dataframe like this:

>>> df = pd.DataFrame(np.array([[1, 2], ['a', 'b']]), columns=['col1', 'col2'])
>>> df
  col1 col2
0    1    2
1    a    b

What I am trying to do is to create a dict from this dataframe where row index is the key and col1 and col2 are a value in a tuple form for my dict. Here is what I am doing but this returns a list:

>>> import numpy as np
>>> import pandas as pd
>>> df.T.to_dict('list')
{0: ['1', '2'], 1: ['a', 'b']}

Here is what I am trying to get:

{0: ('1', '2'), 1: ('a', 'b')}

wwnde · Accepted Answer · 2021-04-28 20:50:49Z

3

Lets try

df.agg(tuple,1).to_dict()

answered Apr 28, 2021 at 20:50

wwnde

26.7k6 gold badges22 silver badges38 bronze badges

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

Comments

BENY · Accepted Answer · 2021-04-28 20:53:42Z

1

Not sure about the speed but with itertuples

dict(zip(df.index, df.itertuples(index=False, name=None)))
{0: ('1', '2'), 1: ('a', 'b')}

Or

dict(zip(df.index,map(tuple,df.values)))

answered Apr 28, 2021 at 20:53

BENY

324k22 gold badges176 silver badges250 bronze badges

Comments

sacuL · Accepted Answer · 2021-04-28 21:04:11Z

1

Another way is to take the dict you have apply a dict comprehension:

>>> {k: tuple(v) for k, v in df.T.to_dict("list").items()}
{0: ('1', '2'), 1: ('a', 'b')}

edited Apr 28, 2021 at 21:04

answered Apr 28, 2021 at 20:51

sacuL

51.6k9 gold badges88 silver badges115 bronze badges

Collectives™ on Stack Overflow

Create dict from a pandas dataframe with values as tuple

3 Answers 3

Comments

Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

3 Answers 3

Comments

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related