开发者

Comma seperated file that I want to load into a dictionary

I have a comma seperated file, a row looks like:

"ABC234234", 2开发者_如何学编程3

I want to load this into a dictionary, with the key being the first part i.e. "ABC234234"

I have to remote the double quotes also.

What's the pythonic way of doing this?


I would suggest (like always) opening the CSV file with the with statement (which ensures it will be closed when you're done!) -- apart from that, @carl's answer is generally fine:

import csv

with open('yourfile.csv', 'rb') as f:
    thedict = dict(csv.reader(f))

and then freely use thedict as you require.

Note that the values (as well, of course, as the keys) will be strings. If you know that the second column always has an integer, and want to have ints as values, you could replace the assignment with

    thedict = dict((k, int(v)) for k, v in csv.reader(f))

or, if you prefer to avoid excessive compactness/density in your code, decompose this latest statement into, for example:

    ks_vs = ((k, int(v)) for k, v in csv.reader(f))
    thedict = dict(ks_vs)

or break it down even further, if you wish, of course.

This works in Python 2.6 or better. If you're stuck with 2.5, to make it work, add

from __future__ import with_statement

at the top of the module -- the rest of my advice still applies;-).


import csv
d = dict(csv.reader(open("foo.txt", "rb")))


You asked for Pythonic. If you wish to follow one of the precepts in the Zen of Python ("Errors should never pass silently") and you want to check that there are no duplicate keys in your data, or do other error checking or sanitising (examples: key can't be an empty string, want to strip leading/trailing whitespace), you'll need to write more detailed code.

#untested example
import csv
with open('the_file.csv', 'rb') as f:
    reader = csv.reader(f)
    the_dict = {}
    for rownum, row in enumerate(reader, start=1):
        if len(row) != 2:
            error('row length is not 2', rownum, row)
            continue
        k, v = [item.strip() for item in row]
        if not k:
            error('key is empty string', ...); continue
        if k in the_dict:
            error(...); continue
        the_dict[k] = v
0

上一篇:

下一篇:

精彩评论

暂无评论...
验证码 换一张
取 消

最新问答

问答排行榜