正則表達式對文本的Python列表解析

Question

我有一個文本文件，其中包含文本中的整數。一行或一行中有一個或多個整數。我想用正則表達式找到這些整數並計算總和。

我已成功地編寫代碼：

import re 
doc = raw_input("File Name:") 
text = open(doc) 
lst = list() 
total = 0 

for line in text: 
    nums = re.findall("[0-9]+", line) 
    if len(nums) == 0: 
     continue 
    for num in nums: 
     num = int(num) 
     total += num 
print total 

，但我也想知道清單理解的版本，有人可以幫忙嗎？

Answer 1

既然您想在找到它們之後計算數字的總和，最好在sum()範圍內使用re.finditer()的生成器表達式。另外，如果文件的大小不是非常大，則最好一次讀取它，而不是一次讀取一行。

import re 
doc = raw_input("File Name:") 
with open(doc) as f: 
    text = f.read() 

total = sum(int(g.group(0)) for g in re.finditer(r'\d+', text))