如何使用唯一代码从CSV文件读取一行？

Question

我有一个系统，用户可以在其中将详细信息写入csv文件，并为该详细信息集生成一个ID。 我希望能够通过输入与该行相对应的ID来从文件中读取一行详细信息。

我在这里看到了一个与我的查询相同的查询，但是我不明白它是如何工作的，我认为我必须为此修改代码。

如果有帮助，这就是写作系统的工作方式：

with open('students.csv', 'a') as csvFile:
    writer = csv.writer(csvFile)

    forename = str(input("Enter student forename: "))
    surname = str(input("Enter student surname: "))
    birthdate = int(input("Enter student date of birth (format as daymonthyear): "))
    address = str(input("Enter student address: "))
    phone_number = int(input("Enter student home phone number: "))
    gender = str(input("Enter student gender: "))
    student_id = random.sample(range(1, 31), 30)

    writer.writerow([forename, surname, birthdate, address, phone_number, gender, student_id])

Answer 1

我认为这对于pandas来说相当容易：

import pandas as pd
csvFile_df = pd.read_csv('students.csv', 
                         header=None, 
                         names=['forename', 'surname', 'birthdate', 'address', 'phone_number', 'gender', 'student_id'])

def return_row_of_details_from_file(df, col, val):
    """ Given pd.DataFrame, `df`, return row of `df` corresponding to
    column `col` having value `val`
    """
    return df[df[col]==val]

# bonus: a curried version of the above which is a ready-made `student_id`-searcher:
return_student_id_details = \
    lambda val, df=csvFile_df, col='student_id': df[df[col]==val]

# example:
details_for_student_7 = return_student_id_details(7)

此函数只是DataFrame pandas索引，并且将返回（子） DataFrame ，该DataFrame由与输入val上匹配的行组成：对您来说， student_id 。 如果csvFile中的student_id值是唯一的，则如果传递给该函数的student_id位于df ，则它将返回1行，7列的pd.DataFrame如果不是，则返回空的pd.DataFrame对象。

上面的确应该有一些错误处理，例如，当col不在df ，除了KeyError之外，还可以进行一些修饰，例如将行作为dict列表返回等，但是，绝招。

Answer 2

正如我在评论中说的那样，没有捷径可做您想要做的事情，因此有必要读取文件并比较每一行，看看其中是否有学生证。

import csv

target_id = input("Enter target student id: ")
# For example: '03ebdf6c-4768-4838-8128-c75ee0a8306d'

with open('students.csv', 'r', newline='') as csvFile:
    for row in csv.reader(csvFile):
        if row[6] == target_id:
            print('Target id found:', row)
            break
    else:
        print('No student with id {} found'.format(target_id))

正如我还提到的那样，使用random.sample()不能保证生成的ID是唯一的。 一种简单的方法是使用内置的uuid模块，如下所示。

使用它生成的学生ID有点长而丑陋，但它们都是唯一的。 只给它们编号可能会更有意义，这样它们可能会更短一些，但是添加新文件的代码会（稍微）更加复杂，因为它将需要先读取整个文件才能找到最后一行的ID。目前在里面。

请注意，在此代码段和上一个代码段中，csv文件都是如何使用newline-''来打开的，该文档已记录了这样做的方式，并且可以避免某些平台上的潜在问题。

import csv
from uuid import uuid4

with open('students.csv', 'a', newline='') as csvFile:
    writer = csv.writer(csvFile)

    forename = input("Enter student forename: ")
    surname = input("Enter student surname: ")
    birthdate = input("Enter student date of birth (format as daymonthyear): ")
    address = input("Enter student address: ")
    phone_number = input("Enter student home phone number: ")
    gender = input("Enter student gender: ")

    student_id = uuid4()  # Generate unique random UUID.

    writer.writerow([forename, surname, birthdate, address, phone_number, gender,
                     student_id])