-1

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

  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*

或者

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!"
4

1 回答 1

0

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

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

您可以采取以下方法：

  1. 打开文件并读取行，按制表符拆分它们并删除换行符。

  2. 创建一个函数，该函数返回其请求的列包含搜索词的第一行。在此示例中，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))
于 2013-03-28T16:33:00.330 回答