Converting a Dictionary with Lists of Tuples into DataFrame in Python

Question

I am attempting to converting a 2 key dictionary with lists of tuples as values into a pandas DataFrame. The data I am working with looks like this:

dictionary={'Week':[(week1,week2),(week3,week4)],'Numbers':[(number1,number2),(number3,number4)]}

Is there anyway to convert this into a pandas DataFrame that looks like:

Week   Numbers
week1  number1
week2  number2
week3  number3
week4  number4

I have tried forcing the dictionary into a DataFrame using pandas.DataFrame(dictionary) but I end up with one row with two very wide columns. Any guidance will be greatly appreciated!

score 4 · Answer 1 · answered Jan 03 '20 at 18:52

4

You can use itertools.chain.from_iterable to flatten the tuples for each keys and call the dataframe constructor:

import itertools
df = pd.DataFrame({k: [*itertools.chain.from_iterable(v)] for k,v in dictionary.items()})

    Week  Numbers
0  week1  number1
1  week2  number2
2  week3  number3
3  week4  number4

answered Jan 03 '20 at 18:52

anky

71,373
8
36
64

1

I keep getting reminded that `itertools.chain.from_iterable()` is so useful! Nice answer! – mechanical_meat Jan 03 '20 at 19:02
1

@bernie thank you . one of the [fastest](https://stackoverflow.com/a/953097/9840637) too :) – anky Jan 03 '20 at 19:05

Andrej Kesely · Accepted Answer · 2020-01-03T19:01:59.833

3

You can use .explode() function (from version 0.25.0+):

dictionary={'Week':[('week1','week2'),('week3','week4')],'Numbers':[('number1','number2'),('number3','number4')]}
df = pd.DataFrame(dictionary)
print(pd.concat([df.explode('Week')['Week'], df.explode('Numbers')['Numbers']], axis=1))

Prints:

    Week  Numbers
0  week1  number1
0  week2  number2
1  week3  number3
1  week4  number4

edited Jan 03 '20 at 19:01

answered Jan 03 '20 at 18:56

Andrej Kesely

118,151
13
38
75

score 3 · Answer 3 · answered Jan 03 '20 at 18:57

Using only list comprehension:

dictionary={'Week':[('week1','week2'),('week3','week4')],'Numbers':[('number1','number2'),('number3','number4')]}
pd.DataFrame({
    k: [t for tup in value for t in tup] for k, value in dictionary.items()
})

Or ndarray.flatten:

pd.DataFrame({
    k: np.array(value).flatten() for k, value in dictionary.items()
})

Converting a Dictionary with Lists of Tuples into DataFrame in Python

3 Answers3