views:

160

answers:

2

Let's say I have a list:

a = ['apple', 'carrot']

and a dictionary:

d ={'apple': [2,4], 'carrot': [44,33], 'orange': [345,667]}

How can I use the list a as a key to lookup in the dictionary d? I want the result to be written to a comma-separated textfile like this

apple,    carrot
2,        44
4,        33

Corrected the a-list from a = ['apple', 'orange'] to a = ['apple', 'carrot']

+5  A: 
a = ['apple', 'orange']
d ={'apple': [2,4], 'carrot': [44,33], 'orange': [345,667]}

print ',\t'.join(a)
for row in zip(*(d[key] for key in a)):
    print ',\t'.join(map(str, row))

Output:

apple,  orange
2,      345
4,      667
Daniel Stutzbach
What does the * in the zip-expression do?
Thanks. I learned something today.williamx - check out http://docs.python.org/library/functions.html#zip
Mark Thalman
@williamx The * makes it so the values are passed as separate arguments to zip(), instead of passing one argument containing all of the values. It's documented here: http://docs.python.org/tutorial/controlflow.html#unpacking-argument-lists
Daniel Stutzbach
+1  A: 

I know other people have been faster and their solutions are similar, but here's my take (take it or leave it):

a = ['apple', 'orange']

d ={'apple': [2,4], 'carrot': [44,33], 'orange': [345,667]}

fo = open('test.csv','w')
fo.write(',\t'.join(a)+'\n')
for y in xrange(len(d[a[0]])):
    fo.write(',\t'.join([str(d[i][y]) for i in a])+'\n')

fo.close()

which generates the file test.csv:

apple,  orange
2,      345
4,      667
Justin Peel