Question

这个问题的基本概要是读取文件，使用re.findall（）查找整数，查找＆＃39; [0-9] +＆＃39;的正则表达式。然后将提取的字符串转换为整数并总结整数。

我已经完成了这个问题，但我想更多地将代码浓缩为两行。

这是我的原始代码：

import re

fh = raw_input("Enter filename: ")
#returns regex_sum_241882.txt as default when nothing is entered
if len(fh)<1 : fh = "regex_sum_241882.txt"
file = open(fh)
sums = list()
#goes through each line in the file
for line in file:
    #finds the numbers in each line and puts them in a list
    nums = re.findall('[0-9]+',line)
    #adds the numbers to an existing list
    for num in nums:
        sums.append(int(num))
#sums the list
print sum(sums)

现在这是我目前的紧凑代码：

import re
lst = list()
print sum(for num in re.findall('[0-9]+',open("regex_sum_241882.txt").read())): int(num))

它没有用，并且给了我 SyntaxError：语法无效

有人能指出我正确的方向吗？我觉得我做同样的事情，但我不确定语法错误是什么。

Answer 1

尝试这种方式：

print sum(int(num) for num in re.findall('[0-9]+', open("regex_sum_241882.txt").read()))

列表理解;将代码压缩为两行

1 个答案: