python - 返回要索引的文本文件中的项目列表

Question

我正在尝试编写一个从文本文件返回城市列表的函数，以便在调用该函数时，您可以在其后面放置一个索引，它将返回相应的城市。

示例:

citylist('MYFILE.txt')[3]

到目前为止，我已经

def citylist(filename):
    assert type(filename)==str

    with open(filename) as FileObject:
        for line in FileObject:
            q=line.split('\t')
            print q[12],

有 500 条投诉。当我将每个字符串(投诉)拆分为一个列表后，城市名称是列表中的第 13 个索引。但我陷入困境，因为我所能做的就是将所有城市名称打印为无法索引的非数据类型。

Answer 1

您可以创建一个列表并返回它:

def citylist(filename):
  assert type(filename)==str  #isinstance(filename,str) is more idiomatic here.

  output = []
  with open(filename) as FileObject:
    for line in FileObject:
      q = line.split('\t')
      output.append(q[12]) 
  return output

或者，更简洁:

def citylist(filename):
    with open(filename) as f:
        return [ line.split('\t')[12] for line in f ]

我使用 list-comprehension 构建了列表在最后一个例子中。