random-emails.py

import csv
import random

DATA = "MOCK_DATA.csv"


def get_emails(data=DATA):
    with open(data) as f:
        reader = csv.reader(f)
        next(reader)
        emails = [row[3] for row in reader]
        return emails


def get_random_emails(emails, number_emails=3):
    return random.sample(emails, k=number_emails)


if __name__ == "__main__":
    all_emails = get_emails()
    rand_emails = get_random_emails(all_emails)
    print(rand_emails)