从 CSV 数据创建字典

Question

我有一个函数，它接受一个 CSV 文件并将其拆分为 3 个值；然后isbn、author 和title 创建一个字典，将isbn 值映射到包含author 和title 的元组。这是我当前的代码：

def isbn_dictionary(filename):
    file = open(filename, 'r')
    for line in file:
        data = line.strip('\n')
        author, title, isbn = data.split(',') 
        isbn_dict = {isbn:(author, title)}
        print(isbn_dict)

问题是目前我可以让它为每个isbn 创建一个字典，但不能为所有这些创建一个字典。我目前的输出是：

{'0-586-08997-7': ('Kurt Vonnegut', 'Breakfast of Champions')}
{'978-0-14-302089-9': ('Lloyd Jones', 'Mister Pip')}
{'1-877270-02-4': ('Joe Bennett', 'So Help me Dog')}
{'0-812-55075-7': ('Orson Scott Card', 'Speaker for the Dead')}

我的输出应该是什么：

{'0-586-08997-7': ('Kurt Vonnegut', 'Breakfast of Champions'),
'978-0-14-302089-9': ('Lloyd Jones', 'Mister Pip'),
'1-877270-02-4': ('Joe Bennett', 'So Help me Dog'),
'0-812-55075-7': ('Orson Scott Card', 'Speaker for the Dead')}

这可能是一个非常简单的问题，但我无法理解它。

Answer 1

使用csv module 进行更轻松、更高效的处理以及字典理解：

import csv

def isbn_dictionary(filename):
    with open(filename, newline='') as infile:
        reader = csv.reader(infile)
        return {isbn: (author, title) for isbn, author, title in reader}

您的代码每行只创建了一个字典，并且只打印了字典。您可能想返回字典。

使用字典推导不仅使函数更紧凑，而且效率更高。字典是在 C 代码中一次性创建的，而不是在 Python 循环中一一添加键和值。

Answer 2

你需要在循环之前声明isbn_dict，像这样：

def isbn_dictionary(filename):
    file = open(filename, 'r')
    isbn_dict = {}
    for line in file:
        data = line.strip('\n')
        author, title, isbn = data.split(',') 
        isbn_dict[isbn] = (author, title)
    print(isbn_dict)

这样，每个项目都会添加到现有字典中。