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 int
s 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
精彩评论