Saving a python dictionary (of lists) in a csv file

Question

A python dictionary (of lists) can be saved in a CSV file as answered in the post available at Write dictionary of lists to a CSV file.

The answer is as follows (directly copied from the above post),

d = {"key1": [1,2,3], "key2": [4,5,6], "key3": [7,8,9]}
with open("test.csv", "wb") as outfile:
   writer = csv.writer(outfile)
   writer.writerow(d.keys())
   writer.writerows(zip(*d.values()))

How can we achieve the same task when the lengths the of lists available in the dictionary are different as follows,

d = {"key1": [1,2,3], "key2": [5], "key3": [6,9]}

So that, it results in the following format (in the CSV),

  key1  key2  key3
   1     5     6
   2           9
   3

@Useless Yep, that's a solution. But, aren't there any other direct ways to achieve this? — cuser
– cuser, Commented Jul 12, 2018 at 11:45
You're currently calling zip. You can change this to itertools.izip_longest. How many other ways do you want to achieve the same thing? You can easily write a loop explicitly, or write your own equivalent version of izip_longest, but ... why do you want to? — Useless
– Useless, Commented Jul 12, 2018 at 11:57

Rakesh · Accepted Answer · 2018-07-12 11:56:41Z

2

If you can use pandas its very simple.

Demo:

import pandas as pd
d = {"key1": [1,2,3], "key2": [5], "key3": [6,9]}
df = pd.DataFrame.from_dict(d, orient='index')
df = df.transpose()
df.to_csv("test.csv", index=False)

answered Jul 12, 2018 at 11:56

Rakesh

82.9k17 gold badges86 silver badges122 bronze badges

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

Collectives™ on Stack Overflow

Saving a python dictionary (of lists) in a csv file

1 Answer 1

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related