예제 #1
0
    ]


if __name__ == "__main__":
    parser = argparse.ArgumentParser(
        description='specify number of email data volume')
    parser.add_argument('volume_number',
                        type=int,
                        nargs='?',
                        help='specify the data volume number')
    args = parser.parse_args()
    if args.volume_number == None:
        args.volume_number = 1
    print("VOLUME NUMBER: ", int(args.volume_number))
    mw = MailWorker()
    data, answereddata = mw.get_message_id_list(int(args.volume_number))
    mails = mw.fetch(data, answereddata)
    mw.mailclose()
    print(len(mails), "mails in dataset")
    mydata = np.empty((len(mails), len(usefulHeaders())), dtype='object')
    row = 0
    for m in mails:
        col = 0
        for k in usefulHeaders():
            mydata[row][col] = m[k]
            col += 1
        row += 1
    print("populated ndarray")

    MyDataFrame = pd.DataFrame(mydata, columns=usefulHeaders(), dtype=str)
    if args.volume_number == 1: