python3x使用类似sql的条件从CSV表中选择一列和一行

Question

这是一个CSV文件kidsList.csv，逗号分隔，第一行是一个标题。第一列Id是唯一的，因此可以将其用作主键。在

Id,Name,Age
A1,Alice,6
A2,Becca,5
B1,Cindy,7

现在我想找到Id为A2的名称；答案应该是“Becca”。在

在SQL中它类似于SELECT Name FROM table WHERE Id = "A2"

如何在python3.x中实现这一点？这个操作非常简单，所以我想使用标准库，比如csv，而不是像{}这样的非标准库。在

Answer 1

您可以使用csv库将csv文件转换为二维列表，然后在数组中循环：

import csv

key = 'A2'
category = 'Name'

with open('kidsList.csv', 'r') as file:
    contents = list(csv.reader(file))

index = contents[0].index(category)

for i in range(1, len(contents)):
    if contents[i][0] == key:
        print(contents[i][index])
        break