在 CSV 中搜索项目

Question

我正在尝试搜索 .csv 文件（我在 Excel 中打开）并在字段中查找特定数字。我正在搜索的数字来自 GUI 中的用户输入。如果在该字段中找到该数字，则将输出同一行中其他字段中的所有项目。这是文件中的内容： screen shot of the file in excel 问题是我似乎无法创建一段可以读取 .csv 并找到数字的代码。 这是我到目前为止所拥有的（这只是代码中不起作用的部分）：

def search(): # defining the function
term=str(e3.get()) # getting the user input and setting it to the varible 'term'
import csv # from all my researching, this is required to open the file
open('Book1.csv') # opens the file
# the code to look through the file would be here. It must search for the number in the correct field and output an error if it can't find it
print() #this will print the data in the same row as the number from the different fields for the user

如果您有解决方案，请给我代码，它可以完全满足我的需要。如果您解释它的作用，我将不胜感激，但如果您不解释也没关系。感谢您提前回复。

Answer 1

您可以使用 python 的 csv 模块这样做：

import csv

def search():
    term = #something
    reader = csv.reader(open('Book1.csv', 'r'))
    for row in reader:
        if row[0] == term:
            return row[1:]
    return None # return None if no match

Answer 2

这里是熊猫解决方案：

让我们从创建示例数据开始：

import io
s = u"""bar_code,item,price
1,Spam,0.1
2,Beans,0.2
3,Egg,0.2
4,Milk,0.3"""

file = io.StringIO(s)

现在是实际代码：

import pandas as pd
df = pd.read_csv(file) 
#df = pd.read_csv('Book1.csv')

lookup = 0.2 # lookup value
matches = df[df['price'] == lookup] # filter rows

# if you find items
if len(matches)>0:
    items = matches.drop('price', axis=1).values.tolist() #drop price column
    print(items)
else:
    print("No match!")

返回：

[[2, 'Beans'], [3, 'Egg']]