Question

Python 文本内容指定字段提取

test.txt 内容如下
1|name1|agg1|add1|phone1|card1|
2|name2|agg2|add2|phone2|card2|

我希望可以用Python提取其中某个字段的内容并保存到另外一个文件。例如我想提取第二个字段。
也就是 "name1" 和 "name2" 并保存到另外一个 output.txt 里。 我应该如何写？

Answer 1

def iterdatainfile(filename, spliter='\t'):
    with open(filename, 'rt') as handle:
        for ln in handle:
            yield ln.split(spliter)

focue, LF = 1, '\n'
with open("output.txt", 'wt') as handle:
    handle.writelines([row[focue] + LF
                       for row in iterdatainfile('test.txt', 
                                                 spliter='|')])