[英]Creating a dictionary from a CSV file
我正在尝试从 CSV 文件中获取输入,然后将其推送为字典格式(我使用的是 Python 3.x)。
我使用下面的代码读取 CSV 文件,并且可以正常工作:
import csv
reader = csv.reader(open('C:\\Users\\Chris\\Desktop\\test.csv'), delimiter=',', quotechar='|')
for row in reader:
print(', '.join(row))
但现在我想将结果放入字典中。 我希望 CSV 文件的第一行用作字典的“键”字段,CSV 文件中的后续行填写数据部分。
样本数据:
Date First Name Last Name Score
12/28/2012 15:15 John Smith 20
12/29/2012 15:15 Alex Jones 38
12/30/2012 15:15 Michael Carpenter 25
我怎样才能让字典工作?
创建一个字典,然后遍历结果并填充字典中的行。 请注意,如果您遇到具有重复日期的行,您将必须决定要做什么(引发异常、替换前一行、丢弃后一行等...)
这是 test.csv:
Date,Foo,Bar
123,456,789
abc,def,ghi
和相应的程序:
import csv
reader = csv.reader(open('test.csv'))
result = {}
for row in reader:
key = row[0]
if key in result:
# implement your duplicate row handling here
pass
result[key] = row[1:]
print(result)
产量:
{'Date': ['Foo', 'Bar'], '123': ['456', '789'], 'abc': ['def', 'ghi']}
或者,使用 DictReader:
import csv
reader = csv.DictReader(open('test.csv'))
result = {}
for row in reader:
key = row.pop('Date')
if key in result:
# implement your duplicate row handling here
pass
result[key] = row
print(result)
结果是:
{'123': {'Foo': '456', 'Bar': '789'}, 'abc': {'Foo': 'def', 'Bar': 'ghi'}}
或者您可能希望将列标题映射到该列的值列表:
import csv
reader = csv.DictReader(open('test.csv'))
result = {}
for row in reader:
for column, value in row.items(): # consider .iteritems() for Python 2
result.setdefault(column, []).append(value)
print(result)
这会产生:
{'Date': ['123', 'abc'], 'Foo': ['456', 'def'], 'Bar': ['789', 'ghi']}
你需要一个 Python DictReader类。 可以从这里找到更多帮助
import csv
with open('file_name.csv', 'rt') as f:
reader = csv.DictReader(f)
for row in reader:
print row
@phil-frost 的帮助非常有帮助,这正是我想要的。
之后我做了一些调整,所以我想在这里分享一下:
def csv_as_dict(file, ref_header, delimiter=None):
import csv
if not delimiter:
delimiter = ';'
reader = csv.DictReader(open(file), delimiter=delimiter)
result = {}
for row in reader:
print(row)
key = row.pop(ref_header)
if key in result:
# implement your duplicate row handling here
pass
result[key] = row
return result
你可以这样称呼它:
myvar = csv_as_dict(csv_file, 'ref_column')
其中 ref_colum 将是每一行的主键。
import csv
def parser_csv(PATH):
reader = csv.reader(open("{}.csv".format(PATH), 'r'))
dict = {}
list_dict = []
counter = 0
for row in reader:
if counter == 0:
first_row = row
ecc = len(first_row)
counter += 1
else:
for col in range(ecc):
dict.update({first_row[col]:row[col]})
list_dict.append(dict)
return list_dict
print(len(parser_csv("path")))
# Have one less csv file (first row is keys of dict)
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.