蟒蛇解析文件

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) 

Answer 2

您可以使用csv模塊（因爲您的數據看起來逗號分隔，至少在你的例子）：

import sys 
import csv 
with open('mail_addresses.txt', 'w') as outfile: 
    for row in csv.reader(open(sys.argv[1], 'rb')): 
     outfile.write("%s\n" % row[1]) 

Answer 3

嘗試是這樣的：

import sys 

Mailfile = sys.argv[1] 
Outfile = sys.argv[2] 

try: 
    in_file = open(Mailfile, 'r') 
    out_file = open(Outfile, 'a') 

    for mail in in_file.readlines(): 
     address = mail.split(',')[1].strip() 
     out_file.write(address+',') #if you want to use commas to seperate the files, else use something like \n to write a new line. 
finally: 
    in_file.close() 
    out_file.close()