帮助在 python 中查找 csv 中的项目

Question

我是Python新手。我有一个不变的 csv，这是一个示例（我只是在 python 控制台中按行打印了它）

['george', 'williams', '277389', 'susan thompson', '2042228888']
['john', 'smith', '833999', 'george smith', '2041118833']
['michael', 'jackson', '281038', 'ronald jackson', '2041128493']

字段标题

['firstname', 'lastname', 'idnumber', 'emergency contact', 'emerg contact ph']

这些是我需要能够输入 id 号的 ，这会启动对 csv 的搜索，并输出个人的名字、姓氏、紧急联系人、电话号码。有什么想法吗？我真的需要知道从哪里开始，即我是否应该将 csv 的内容读入字典中

Answer 1

我个人会选择字典：

records = [
    ['george', 'williams', '277389', 'susan thompson', '2042228888'],
    ['john', 'smith', '833999', 'george smith', '2041118833'],
    ['michael', 'jackson', '281038', 'ronald jackson', '2041128493'],
    ]

from operator import itemgetter
recordsbyid = dict(zip(map(itemgetter(2),records),records))

然后您可以执行

>>> recordsbyid['277389']
['george', 'williams', '277389', 'susan thompson', '2042228888']

itemgetter 选择第二个元素（id），map 适用于每条记录，zip将 ids 与其记录连接成由 (id,record) 组成的元组列表。 dict 将其转换为字典。

Answer 2

我会将此 csv 转换为 sqlite 并使用查询：

SELECT * FROM data WHERE idnumber = %s

我喜欢使用数据库中组织的数据，它可能会在未来为您带来更多优势（更复杂的查询）。

要将 cvs 转换为 sqlite 并测试查询，请使用 适用于 Firefox 的 SQLite Manager 插件< /a>.

Answer 3

如果您需要扫描文件并仅查找所需的行一次，则无需将所有数据转换为字典 - 只需逐行读取行，直到找到您的行：

import csv

def find_row_by_id(filename, key_column, id):
    with f = open(filename, 'rb'):
        my_reader = csv.reader(f)
        for row in my_reader:
            if row[key_column] == id:
                return row
    raise Error("Could not find row")

print find_by_row('eggs.csv', 2, my_id) # my_id should by a string

如果您的文件很小并且您需要仅通过 id 进行多次搜索，按照其他答案的建议将其转换为字典。

另一方面，如果您的文件非常（非常）大并且您需要进行快速查找和/或多次查找，请首先将您的 csv 文件读入键值数据库： 适用于 Linux 的可靠且高效的键值数据库？

Answer 4

使用 raw_input 获取所需的 id。

当你拥有它时，使用 if 语句来查看 id 是否匹配：

Where line is ['george', 'williams', '277389', 'susan thompson', '2042228888']

if line[2] == id:
    for x in line:
        if not x == line[2]:
            print x

希望它有帮助！

Answer 5

让我们首先将每一行制作成一个字典，使用字段标题作为键，使用字段数据作为值。我们想要一个字典，其中每个键值对都来自我们通过将这两个列表“压缩”在一起而获取的对：dict(zip(field_titles, row))。我们可以使用列表理解从输入行列表中创建这些字典的列表：[dict(zip(field_titles, row)) for row in data]。

我们希望能够通过 id_number 查找这些内容，因此我们将创建一个包含字典，其中键是 idnumber，值是行字典。我们可以通过在 row_dict 中查找来获取 idnumber：dict((row_dict['id_number'], row_dict) for row_dict in data)。