在嵌套列表中搜索值

Question

我有一个数据库文件，csv 文件。我想在这个列表中搜索一个名字。要知道它是否存在。如果不显示要添加的选项。名称旁边的数字是客户编号。如何在此列表中搜索值？

lines=[]
user_name=input("PLEASE ENTER YOUR FIRST AND LAST NAME: ")
with open("data_base.csv", "r") as in_file:
    for line in in_file:
        row = line.split(",")
        for i in range(1, len(row)):
            row[i] =eval(row[i])
            lines.append(row)

如果我打印列表，结果是下一个：

[['Mariana Ortigoza', 1], ['James Butt', 2], ['Josephine Darakjy', 3], ['Art Venere', 4], ['Lenna Paprocki', 5], ['Donette Foller', 6], ['Simona Morasca', 7], ['Mitsue Tollner', 8], ['Leota Dilliard', 9], ['Sage Wieser', 10]]

Answer 1

一种方法是使用operator.itemgetter 提取每个子列表的第一个元素。

然后测试输入的名字是否在set的名字中；如果没有，则附加到lines。

from operator import itemgetter

lines = [['Mariana Ortigoza', 1], ['James Butt', 2], ['Josephine Darakjy', 3],
         ['Art Venere', 4], ['Lenna Paprocki', 5], ['Donette Foller', 6],
         ['Simona Morasca', 7], ['Mitsue Tollner', 8], ['Leota Dilliard', 9],
         ['Sage Wieser', 10]]

user_name = input("PLEASE ENTER YOUR FIRST AND LAST NAME: ")

if user_name not in set(map(itemgetter(0), lines)):
    lines.append([user_name, len(lines)+1])