Customs_day6

with open ('Day6.txt', 'rt') as f:
    data = f.read().strip()
data = data.split('\n\n')

# part 1
total_yes = 0
remove = set(['\n'])
for group in data:
    yes_answers = list(set(group).difference(remove))
    total_yes += len(yes_answers)
print(total_yes)

# part 2
common_yes = 0
for group in data:
    for ind, traveler in enumerate(group.split()):
#         print('set',set(traveler))
        if ind == 0:
            yes_answers = set(traveler)
        yes_answers = yes_answers.intersection(set(traveler))
    common_yes += len(yes_answers)
print(common_yes)