时间戳

首页 > 解决方案 > 如何将 CSV 文件中的行存储在字典中？

python - 如何将 CSV 文件中的行存储在字典中？

问题描述

我必须用 Python 从 CSV 文件中提取索引 1 和 5 处的所有行，并将它们存储在字典“名称”中。但是，由于我是 Python 新手，我真的很努力完成这个任务。这是我的代码：

names = {}
with open("../data/names.csv") as file:
    for line in file:
        print(line)

输出：

Id,Name,Year,Gender,State,Count
1,Mary,1910,F,AK,14
2,Annie,1910,F,AK,12
3,Anna,1910,F,AK,10
4,Margaret,1910,F,AK,8
5,Helen,1910,F,AK,7
6,Elsie,1910,F,AK,6
...

如果我尝试将索引 1（字符串“Name”）和索引 5（int）存储在字典“names”中，则会得到错误的输出。这里的代码：

names = {}
with open("../data/names.csv") as file:
    for line in file:
        names = {"name": line[1], "count": line[5]}
print(names)

（错误的）输出：

{'name': '6', 'count': '2'}

标签： pythoncsvdictionary

解决方案

未指定所需的输出数据结构，我认为问题是关于列（而不是行）

with open('sample.csv','r') as file_handle:
    file_content = file_handle.read()

list_of_dicts = []
for index, line in enumerate(file_content.split('\n')):
    line_as_list = line.split(',')
    if index==0:
        pass # ignore the headers
    else:
        line_dict = {'name': line_as_list[1], 'count': line_as_list[5]}
        list_of_dicts.append(line_dict)
print(list_of_dicts)

如果所需的数据结构是单个字典，其中包含每行的条目列表，

results = {'name': [], 'count': []}
for index, line in enumerate(file_content.split('\n')):
    line_as_list = line.split(',')
    if index==0:
        pass # ignore the headers
    else:
        results['name'].append(line_as_list[1])
        results['count'].append(line_as_list[5])
print(results)

推荐阅读