Question

我有一个以制表符分隔的文本文件（每列都负责某个组件。）：

  1 (who?)        2 (age)        3 (Does what?)        4 )(where lives?)
A    Dog             4                 Walks           Lives in a box
B   Parrot           2                 Eats            Lives in a cage
C   Cat              3                 Sleeps          Lives in a house

用户必须在两种选择中选择查找有关动物的所有信息。用户可以在第一列或第四列中输入他的猜测：

1=Cat*

OR

4=Live in a box**

如果有匹配则打印整行。

在第一种情况下，如果用户决定使用第一列（谁？）来查找有关动物的信息，则会显示：
```
Cat              3                 Sleeps          Lives in a house
```
在第二种情况下，如果用户决定使用第四列（生命？）来查找有关该动物的信息，将显示：
```
Dog             4                 Walks           Lives in a box
```

我需要的是找到一个能够搜索动物的正确功能。我尝试使用search（）和match（），但无济于事。

                        **EDIT after codesparkle's advice**

我似乎无法将建议的功能集成到我的程序中。我对line [column-1]位特别困惑。我尝试用这个功能“玩”但我无法想出任何有用的东西。所以这是我需要完成的真正的代码。

while True:
    try:
        OpenFile=raw_input(str("Please enter a file name: ")) 
        infile=open(OpenFile,"r")
        contents=infile.readlines()
        infile.close()

        user_input = raw_input(str("Enter A=<animal> for animal search or B=<where lives?> for place of living search: \n"))

        if user_input.startswith("A="):
            def find_animal(user_input,column):
                return next(("\t".join(line) for line in contents
                             if line[column-1]==user_input),None)
            find_animal(user_input[1:]) 
            print str((find_animal(user_input[1:], "WHO?"))) # I messed up the first time and gave numeric names to the columns. In reality they should have been words that indicate the proper name of the column. In this case it is WHO- the first column which signifies the animal.

        else:
            print "Unknown option!"


    except IOError:
        print "File with this name does not exist!"

Answer 1

假设您的输入看起来像这样（您发布的示例包含空格，而不是标签）：

Dog 4   Walks   Lives in a box
Parrot  2   Eats    Lives in a cage
Cat 3   Sleeps  Lives in a house

您可以采取以下方法：

打开文件并读取行，按标签拆分并删除换行符。
创建一个函数，返回请求列包含搜索词的第一行。在此示例中，如果未找到任何内容，则返回None。

with open('input.txt') as input_file:
    lines = [line.rstrip().split('\t')
             for line in input_file.readlines()]

def find_animal(search_term, column):
    return next(('\t'.join(line) for line in lines
                 if line[column-1] == search_term), None)

print(find_animal('Cat', 1))
print(find_animal('Lives in a box', 4))

Python：在包含多个列的文本文件中搜索特定字符串

1 个答案: