Python:通过字典重构有序的文本文件

时间:2012-07-11 18:08:43

标签: python text dictionary key

我有一个我想重组的文本文件。该文件包含功能,功能说明(选项卡分隔,描述符数量会有所不同)以及显示该功能的人员列表。它看起来像这样:

Feature1    detail1    detail2
Person1
Person2
Feature2    detail1    detail2    detail3
Person3

...而且我只想对其进行重组,以便每行有一个特征,并且人们在描述符之后添加到行上,所以它看起来像这样:

 Feature1    detail1    detail2    Person1    Person2
 Feature2    detail1    detail2    detail3    Person3

我想使用Python字典。我的代码将每个功能作为键读取并将详细信息附加为值,但我无法将人员添加为值。有人可以帮忙吗?

import re
import sys
import csv

def reformat(filename):
    mydict = dict()
    for line in filename:
        m = re.search("\AFeature",line[0])  
        if m:
            if str(line) in mydict:
                mydict[line].append(line[0])
            else:
                mydict[line[0]] = (line[1:-1])
    print(mydict)

thefilename = "path"
path_reader=csv.reader(open(thefilename), delimiter = "\t")
rv = reformat(path_reader)

编辑:固定代码缩进

1 个答案:

答案 0 :(得分:1)

我只更改了if ... else部分:

def reformat(filename):
    mydict = dict()
    feature = ''
    for line in filename:
        m = re.search("\AFeature",line[0])  
        if m:
            feature = line[0]
            mydict[feature] = (line[1:-1])
        else:
            mydict[feature].append(line[0])
print(mydict)