Python：文件内容到字典

Question

编写一个函数，接受一个 CSV 文件的文件名（字符串） 包含有关学生姓名和成绩的信息 四门课程并返回信息字典。的钥匙 字典应该是学生的名字和值 应该是他们成绩的浮点数列表。为了 例如，如果文件的内容如下所示：

Mark,90,93,60,90
Abigail,84,50,72,75
Frank,46,83,53,79
Yohaan,47,77,74,96

那么你的函数应该返回一个字典，例如：

out_dict = {'Frank': [46.0, 83.0, 53.0, 79.0],
            'Mark': [90.0, 93.0, 60.0, 90.0],
            'Yohaan': [47.0, 77.0, 74.0, 96.0],
            'Abigail': [84.0, 50.0, 72.0, 75.0]}

这是我的代码：

def dict_from_file (file_name):
    file_pointer = open(file_name, 'r')
    data = file_pointer.readlines()
    print (data)
    output_dict = {}
    for line in data:

    file_pointer.close()
    return (output_dict)

#Main Program
file_name = input("Enter the exact file name with its extension (.i.e., .txt): ")
result = dict_from_file (file_name)
print (result)

如您所见，for 循环中缺少语句。问题是我找不到任何逻辑来首先从文件中获取输入并将其粘贴到字典中。如果我打算提取每行一次，我将如何将其添加到字典中，名称为键，四个数字为值？

Answer 1

严格关注您的代码：

如果你打印出你读过的东西，你会发现你有这个：

['Mark,90,93,60,90\n', 'Abigail,84,50,72,75\n', 'Frank,46,83,53,79\n', 'Yohaan,47,77,74,96']

因此，当您遍历列表中的每个项目时，为使此示例简单，您几乎可以提取第一个项目，这现在将成为您的键，然后取出列表中不包括名称的部分并将每个项目转换为浮点数。

所以，你几乎正在寻找：

l = line.split(',')
output_dict[l[0]] = map(float, l[1:])

上面发生的情况是，您正在通过拆分“，”来获取您正在迭代的每个项目，使其成为一个列表。然后，您将获取列表中的第一项并将其作为您的密钥。然后要分配您的值，您必须使用map 方法，该方法会将您的list 中的每个项目映射到一个浮点数。

您将为正在读取的每个 line 执行该操作。

所以，最后，你将拥有：

def dict_from_file (file_name):
    file_pointer = open(file_name, 'r')
    data = file_pointer.readlines()
    print (data)
    output_dict = {}
    for line in data:
        l = line.split(',')
        output_dict[l[0]] = map(float, l[1:])
    file_pointer.close()
    return output_dict

#Main Program
result = dict_from_file ('stuff.txt')
print (result)

运行该代码，将为您提供：

{'Frank': [46.0, 83.0, 53.0, 79.0], 'Yohaan': [47.0, 77.0, 74.0, 96.0], 'Abigail': [84.0, 50.0, 72.0, 75.0], 'Mark': [90.0, 93.0, 60.0, 90.0]}

Answer 2

我建议你使用 csv 阅读器：

import csv
with open(file_name, 'rb') as csvfile:
    csv_reader = csv.reader(csvfile)
    for row in csv_reader:
        output_dict[row[0]] = [float(n) for n in row[1:]]

如果正确插入你的 for 循环，这样的东西应该可以工作。我还没有测试过代码，但应该不会太远。