python解析文件

Question

我有一个包含用户名和电子邮件的文件,格式如下:

pete,pbellyer@gmail.com

我只想保留电子邮件,所以我想要使用这样的正则表达式:

import re,sys

Mailfile = sys.argv[1]

file = open(Mailfile, "r")

for MAIL in file.readlines():
   tmp = re.split("\n+", MAIL)
   m = re.match( ',(.+)', MAIL)
   m.group(0)

但后来我不知道如何将结果存储在文件中.我总是在新文件中获取最后一个电子邮件地址.

将结果存储在文件中的最佳方法是什么？谢谢!

Answer 1

import sys

infile, outfile = sys.argv[1], sys.argv[2]

with open(infile) as inf, open(outfile,"w") as outf:
    line_words = (line.split(',') for line in inf)
    outf.writelines(words[1].strip() + '\n' for words in line_words if len(words)>1)