Python 使用列表搜索 function in.csv

Question

我有一个.csv 文件分为四个不同的列，我会在 python 中编写一个搜索 function，它给了我打印文档的某些特定行的最厚搜索列表，但要做。 .那是因为如果我只是打开 csv 文档，搜索也可以在内部找到相似的词 - 例如 - 在名称或其他词中（因此，如果我尝试搜索“App”，因为它在列中，结果可能是“App”这个词的确切行，但也是名称的确切行 - 例如随机给出 - 'Applebaum'......我不想要那个）。 在我的脚本中，我曾经为 input() 提供一个变量，但我需要使用 tkinter 所以我认为我必须使用 Entry 而不是 input()... 最后，如果 csv 中的条目不相等，我会这样做文件（与列表匹配），一个消息框给出一条消息，说明找不到。

所以，我尝试了：

import csv
from tkinter import messagebox

def search_parameter():
    accepted_word_for_foods = ['bread', 'jam', 'cake', 'wakamole', 'tomato', 'pear']
    accepted_word_for_drinks = ['water', 'juice', 'milk', 'tea' ]
    accepted_word_for_colors = ['red', 'brown','blue']
    
    param = input('Insert here your param...\n')
    csv_f = csv.reader(open('random_list_example.csv', 'r'))
    
    for row in csv_f:
        if param in row[0] == accepted_word_for_foods:
            print(row)
            
        if param in row[1] == accepted_word_for_drinks:
            print(row)
            
        if param in row[2] == accepted_word_for_colors:
            print(row)
            
        if param != accepted_word_for_foods:
            messagebox.showwarning(title='Not found', message='Param not present or bad digit')
            
        if param != accepted_word_for_drinks:
            messagebox.showwarning(title='Not found', message='Param not present or bad digit')

        if param != accepted_word_for_colors:
            messagebox.showwarning(title='Not found', message='Param not present or bad digit')


search_parameter()

显然，它不起作用。 我也尝试了不同的组合（如果 params == accepted_words_for_foods in row[0]...etc 等等），但我认为问题出在我引用列表的方式上。 但我真的不明白该怎么做。

Answer 1

谢谢你。 这就是我一直在寻找的。 我知道这很简单，但是 python 语法的某些内容对我来说仍然有点误导！

再次感谢！

Answer 2

你正朝着正确的方向前进。

一些建议的更改以解决您遇到的问题：

1. 在费心搜索 csv 之前先检查用户输入是否可以接受
2. 而不是if param in row[0] just do for param == row[0]这解决了重复的代码问题，也解决了你的"app" != "applebaum"问题
3. 因为您想查看每个列检查是否也是该列的接受词，所以您需要一个and

---

import csv
from tkinter import messagebox

def search_parameter():
    accepted_word_for_foods = ['bread', 'jam', 'cake', 'wakamole', 'tomato', 'pear']
    accepted_word_for_drinks = ['water', 'juice', 'milk', 'tea' ]
    accepted_word_for_colors = ['red', 'brown','blue']
    
    param = input('Insert here your param...\n')
    
    
    #check to see if the input is acceptable before bothering with the csv
    if  param not in accepted_word_for_colors and  param not in accepted_word_for_drinks and  param not in accepted_word_for_foods:
         messagebox.showwarning(title='Not found', message='Param not present or bad digit')
    else:
        #param is acceptable, lets look at the csv
        csv_f = csv.reader(open('random_list_example.csv', 'r'))
        for row in csv_f:
            if param == row[0] and param in accepted_word_for_foods:
                print(row)
            if param == row[1] and param in accepted_word_for_drinks:
                print(row)
            if param == row[2] and param in accepted_word_for_colors:
                print(row)
search_parameter()

总体建议的更改。

1. 考虑使用列表列表来管理您接受的单词。 这会将所有接受的单词放入一个变量中。 每个内部列表的索引将对应于 csv 中的列。
2. 创建一个标志/布尔变量来跟踪是否找到该项目。 然后只需在处理结束时检查该标志以向用户弹出错误消息，表明该项目未找到（或不可接受）。

这看起来像：

import csv
from tkinter import messagebox

def search_parameter():
    accepted_words=[['bread', 'jam', 'cake', 'wakamole', 'tomato', 'pear']
                   ,['water', 'juice', 'milk', 'tea' ]
                   ,['red', 'brown','blue']]
    
    param = input('Insert here your param...\n')
    
    #new flag/boolean variable to track if we found the item or not
    word_found = False    
    
    csv_f = csv.reader(open('random_list_example.csv', 'r'))
    
    for row in csv_f:
        #loop through each column using `range()`
        for col in range(3):
            #since we are looping through the columns now, we only need one line of logic to test each column
            if param == row[col] and param in accepted_words[col]:
                word_found = True
                print(row)
    #did that flag get flipped? If not, message the user
    if not word_found:
         messagebox.showwarning(title='Not found', message='Param not present or bad digit')

search_parameter()