如何在Python中使用defaultdict读取行？

Question

我有一个字典，它搜索一个ID名称并在它后面读取令牌。但是我想知道是否有办法读取并打印出包含该ID名称的整行。

这是我到目前为止有：

lookup = defaultdict(list) 
wholelookup =defaultdict(list) 
mydata = open('summaryfile.txt') 

for line in csv.reader(mydata, delimiter='\t'): 
    code = re.match('[a-z](\d+)[a-z]', line[-1], re.I) 
    if code: 
     lookup[line[-2]].append(code.group(1)) 
     wholelookup[line[-2]].append(code.group(0)) 

Answer 1

代码调用csv.reader()这将返回整行的解析版本。在我的测试中，这返回一个值列表。如果这个值列表可以用于“整行”，那么你可以保存它。

你有一行你可以追加一个叫wholelookup的东西。我想你只想在那里保存line而不是code.group(0)。 code.group(0)返回与正则表达式匹配的所有内容，这与line[-1]相同。

所以也许把这一行代码：

wholelookup[line[-2]].append(line) 

或者，也许你需要的值从line联合起来做一个字符串：

s = ' '.join(line) 
wholelookup[line[-2]].append(s) 

如果你想整条生产线，而不是解析的版本，然后做这样的事情：

lookup = defaultdict(list) 
wholelookup = defaultdict(list) 
pat = re.compile('[a-z](\d+)[a-z]', re.I) 

with open('summaryfile.txt') as mydata: 
    for s_line in mydata: 
     values = s_line.split('\t') 
     code = re.match(pat, values[-1]) 
     if code: 
      lookup[values[-2]].append(code.group(1)) 
      wholelookup[values[-2]].append(s_line) 

此为例e预先编制了轻微速度优势的模式。

Answer 2

如果你有足够的内存，最简单的方法是简单地保存在另一个defaultdict行：

wholeline = defaultdict(list) 
... 
idname = line[-2] 
wholeline[idname].append(line)