Python poiFlagEmail示例

示例#1

文件： studentMain.py 项目： yurayli/udacity-ml-class

import sys
import zipfile
from poi_flag_email import poiFlagEmail, getToFromStrings

data_dict = {}

with zipfile.ZipFile('emails.zip', "r") as z:
    z.extractall()

for email_message in os.listdir("emails"):
    if email_message == ".DS_Store":
        continue
    message = open(os.getcwd() + "/emails/" + email_message, "r")

    to_addresses, from_addresses, cc_addresses = getToFromStrings(message)
    to_poi, from_poi, cc_poi = poiFlagEmail(message)

    for recipient in to_addresses:
        # initialize counter
        if recipient not in data_dict:
            data_dict[recipient] = {"from_poi_to_this_person": 0}
        # add to count
        if from_poi:
            data_dict[recipient]["from_poi_to_this_person"] += 1

    message.close()

for item in data_dict:
    print item, data_dict[item]

示例#2

文件： studentMain.py 项目： DingChiLin/FCH808.github.io

import sys
import zipfile
from poi_flag_email import poiFlagEmail, getToFromStrings

data_dict = {}

with zipfile.ZipFile('emails.zip', r) as z
    z.extractall()

for email_message in os.listdir(emails)
    if email_message == .DS_Store
        continue
    message = open(os.getcwd()+emails+email_message, r)
    to_addresses, from_addresses, cc_addresses = getToFromStrings(message) 
    
    to_poi, from_poi, cc_poi = poiFlagEmail(message)
    
    for recipient in to_addresses
        if recipient not in data_dict
            data_dict[recipient] = {from_poi_to_this_person0}
        else
            if from_poi
                data_dict[recipient][from_poi_to_this_person] += 1

    message.close()

for item in data_dict
    print item, data_dict[item]
    
#######################################################    
def submitData()