Python Solution to Advent of Code Day 01

aoc_2020_01.py

import numpy as np

with open("input01.txt") as f:
    a = f.read()
data = np.array([int(n) for n in a.split("\n")[:-1]])

def first(data):
    for i, number in enumerate(data):
        for number2 in data[i:]:
            if number + number2 == 2020:
                print(f"Part 1: {number*number2}")
            for number3 in data[i+1:]:
                if number + number2 + number3 == 2020:
                    print(f"Part 2: {number*number2*number3}")