0

所以我有一个包含以下格式信息的文本文件:

|NAME|70|AST|TYPE|LOCATION|3||MNUM_1|NUM_2|DESCRIPTION|

我的代码在文本文件中搜索 NAME 属性,然后返回除拆分字符“|”之外的所有信息。这是我的代码:

    with open('TEXT.txt') as fd:
    input=[x.strip('|').split('|') for x in fd.readlines()]
    to_search={x[0]:x for x in input}
    print('\n'.join(to_search[name]))

有没有办法编辑我的代码,以便我可以在文本文件中搜索名称属性并选择我想要返回的属性并返回该属性?

4

2 回答 2

0

将第一行作为字段名称怎么样,然后您可以执行以下操作:

search_key = 'NAME'
search_query = 'foo'

with open('TEXT.txt') as fin:
    keys = fin.readline().strip('|').split('|')
    for line in fin:
        columns = line.strip().strip('|').split('|')
        obj = dict(zip(keys, columns))
        if obj[search_key] == search_query:
            ...

正是出于这个原因,许多 csv 文件都遵循此约定(在第一行中命名列)。

于 2013-03-28T16:40:05.380 回答
0

对于这种类型的数据文件最好使用csv包,但在现有代码中您可以这样做:

columns = ['NAME', '70', 'AST', 'TYPE', 'LOCATION', '3', 'SOMETHING', 'MNUM_1', 'NUM_2', 'DESCRIPTION']
return to_search[requested_name][columns.index(requested_column)]
于 2013-03-28T16:36:44.997 回答